NonZero inaccuracies

[attila-dusnoki-htec]

Failing tests:

• test_nonzero_example_cpu

[attila-dusnoki-htec]

FAIL: test_nonzero_example_cpu (__main__.OnnxBackendNodeModelTest)

``` ====================================================================== FAIL: test_nonzero_example_cpu (__main__.OnnxBackendNodeModelTest) ---------------------------------------------------------------------- Traceback (most recent call last): File "/usr/local/lib/python3.8/dist-packages/onnx/backend/test/runner/__init__.py", line 290, in device_test_func return test_func(*args, device=device, **kwargs) File "/usr/local/lib/python3.8/dist-packages/onnx/backend/test/runner/__init__.py", line 467, in run self.assert_similar_outputs( File "../test/py/onnx_backend_test.py", line 59, in assert_similar_outputs np.testing.assert_allclose(ref_outputs[i], File "/usr/local/lib/python3.8/dist-packages/numpy/testing/_private/utils.py", line 1530, in assert_allclose assert_array_compare(compare, actual, desired, err_msg=str(err_msg), File "/usr/local/lib/python3.8/dist-packages/numpy/testing/_private/utils.py", line 763, in assert_array_compare raise AssertionError(msg) AssertionError: Not equal to tolerance rtol=0.001, atol=1e-05 Program = module: "main" condition = @param:condition -> bool_type, {2, 2}, {2, 1}, target_id=0 @1 = nonzero(condition) -> int64_type, {2, 4}, {4, 1}, target_id=0 @2 = @return(@1), target_id=0 Compiled program = module: "main" @0 = check_context::migraphx::gpu::context -> float_type, {}, {}, target_id=0 @1 = hip::hip_allocate_memory[shape=int8_type, {96}, {1},id=main:scratch] -> int8_type, {96}, {1}, target_id=0 condition = @param:condition -> bool_type, {2, 2}, {2, 1}, target_id=0 @3 = load[offset=64,end=68](@1) -> bool_type, {2, 2}, {2, 1}, target_id=0 @4 = hip::copy_to_gpu(condition,@3) -> bool_type, {2, 2}, {2, 1}, target_id=0 @5 = load[offset=0,end=64](@1) -> int64_type, {2, 4}, {4, 1}, target_id=0 @6 = gpu::nonzero(@4,@5) -> int64_type, {2, 4}, {4, 1}, target_id=0 @7 = hip::copy_from_gpu(@6) -> int64_type, {2, 4}, {4, 1}, target_id=0 @8 = hip::sync_stream(@7) -> int64_type, {2, 4}, {4, 1}, target_id=0 @9 = @return(@8), target_id=0 (shapes (2, 3), (2, 4) mismatch) x: array([[0, 1, 1], [0, 0, 1]]) y: array([[0, 1, 1, 0], [0, 0, 1, 0]], dtype=int64) ```

[gyulaz-htec]

This is caused by NonZero's output doesn't have a dynamic shape. This is a known issue in AMDMIGraphX: https://github.com/ROCmSoftwarePlatform/AMDMIGraphX/issues/948 Other related issue is: https://github.com/ROCmSoftwarePlatform/AMDMIGraphX/issues/1886

[gyulaz-htec]

Moving this to blocked because of the dynamic shape support.