Unverified Commit 1a0dc5c3 authored by Aarni Koskela's avatar Aarni Koskela Committed by GitHub
Browse files

test_nvidia_transform: fix variable reference (#1000)

`out_order` is the global parametrization list, not the test fixture argument
parent b90db7ea
......@@ -613,9 +613,9 @@ names = ["dim1_{}_dim2_{}_dim3_{}_dims_{}_dtype_{}_orderA_{}_orderOut_{}_transpo
@pytest.mark.parametrize("dim1, dim2, dim3, dims, dtype, orderA, orderOut, transpose",values,ids=names)
def test_nvidia_transform(dim1, dim2, dim3, dims, dtype, orderA, orderOut, transpose):
if dims == 3 and out_order != "col32":
if dims == 3 and orderOut != "col32":
return
if dtype == torch.int32 and out_order != "col32":
if dtype == torch.int32 and orderOut != "col32":
return
try:
func = F.get_transform_func(dtype, orderA, orderOut, transpose)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment