Update scan.py to fix pascal error.

rankaiyx · web-flow · commit 3e3a099b44ca · 2025-07-03T15:16:55.000+08:00
diff --git a/python/tvm/topi/gpu/scan.py b/python/tvm/topi/gpu/scan.py
@@ -40,6 +40,8 @@ def _can_use_scan_thrust(binop):
     target = tvm.target.Target.current()
     if target is None:
         return False
+    if target.arch == "sm_60":
+        return False
     return binop == tvm.tir.generic.add and any(
         [
             can_use_thrust(target, "tvm.contrib.thrust.sum_scan"),

Original file line number	Diff line number	Diff line change
`@@ -40,6 +40,8 @@ def _can_use_scan_thrust(binop):`
`40`	`40`	`target = tvm.target.Target.current()`
`41`	`41`	`if target is None:`
`42`	`42`	`return False`
	`43`	`+ if target.arch == "sm_60":`
	`44`	`+ return False`
`43`	`45`	`return binop == tvm.tir.generic.add and any(`
`44`	`46`	`[`
`45`	`47`	`can_use_thrust(target, "tvm.contrib.thrust.sum_scan"),`