kazum commented on a change in pull request #5857:
URL: https://github.com/apache/incubator-tvm/pull/5857#discussion_r445411686
##########
File path: tests/python/relay/test_op_level5.py
##########
@@ -270,8 +270,8 @@ def verify_get_valid_counts(dshape, score_threshold,
id_index, score_index):
intrp = relay.create_executor("debug", ctx=ctx, target=target)
out = intrp.evaluate(func)(np_data)
tvm.testing.assert_allclose(out[0].asnumpy(), np_out1, rtol=1e-3,
atol=1e-04)
- # get_valid_count for cuda doesn't do data rearrangement
- if target == 'cuda':
+ # get_valid_count for cuda, opencl doesn't do data rearrangement
+ if target in ['cuda', 'opencl']:
return
Review comment:
Returning here looks wrong to me. The test in the below link doesn't
work for OpenCL too because we don't do data rearrangement for GPU nms
implementation.
https://discuss.tvm.ai/t/nms-compile-fails-for-cuda-target-but-works-fine-for-llvm-target/7045/2
Probably, we should fix non_max_suppression for GPU first?
----------------------------------------------------------------
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]
