masahi commented on pull request #7611:
URL: https://github.com/apache/tvm/pull/7611#issuecomment-796019880
On VK, it fails on (122640, 1) workload. rocm works fine.
```
Traceback (most recent call last):
File "vk_test.py", line 146, in <module>
test_argsort()
File "vk_test.py", line 94, in test_argsort
verify_argsort((1, 122640), axis=1, is_ascend=False, dtype=dtype,
is_dyn=is_dyn)
File "vk_test.py", line 86, in verify_argsort
tvm.testing.assert_allclose(op_res.asnumpy(), ref_res.astype(dtype),
rtol=1e-5)
File "/home/masa/projects/dev/tvm/python/tvm/testing.py", line 82, in
assert_allclose
np.testing.assert_allclose(actual, desired, rtol=rtol, atol=atol,
verbose=True)
File
"/home/masa/anaconda3/lib/python3.8/site-packages/numpy/testing/_private/utils.py",
line 1532, in assert_allclose
assert_array_compare(compare, actual, desired, err_msg=str(err_msg),
File
"/home/masa/anaconda3/lib/python3.8/site-packages/numpy/testing/_private/utils.py",
line 846, in assert_array_compare
raise AssertionError(msg)
AssertionError:
Not equal to tolerance rtol=1e-05, atol=1e-07
Mismatched elements: 122633 / 122640 (100%)
Max absolute difference: 122388
Max relative difference: 84808.
x: array([[116254, 14451, 111687, ..., 64552, 122405, 122011]],
dtype=int32)
y: array([[116254, 14451, 111687, ..., 14845, 100388, 53504]],
dtype=int32)
```
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
