diff --git a/python/tvm/topi/gpu/scan.py b/python/tvm/topi/gpu/scan.py index 5be4033e4575..a4019403bb6e 100644 --- a/python/tvm/topi/gpu/scan.py +++ b/python/tvm/topi/gpu/scan.py @@ -40,6 +40,9 @@ def _can_use_scan_thrust(binop): target = tvm.target.Target.current() if target is None: return False + # Excluding Pascal architecture, use fallback path code + if target.arch in ("sm_60", "sm_61"): + return False # pylint: disable=comparison-with-callable return binop == tvm.tir.generic.add and any( [