We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e3a099 commit ecd8615Copy full SHA for ecd8615
python/tvm/topi/gpu/scan.py
@@ -42,6 +42,7 @@ def _can_use_scan_thrust(binop):
42
return False
43
if target.arch == "sm_60":
44
45
+ # pylint: disable=comparison-with-callable
46
return binop == tvm.tir.generic.add and any(
47
[
48
can_use_thrust(target, "tvm.contrib.thrust.sum_scan"),
0 commit comments