Skip to content

Commit ba7b39d

Browse files
authored
Add Pascal workaround for MoE
apache/tvm#18109
1 parent 424aaa4 commit ba7b39d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

python/tvm/topi/gpu/scan.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,9 @@ def _can_use_scan_thrust(binop):
4040
target = tvm.target.Target.current()
4141
if target is None:
4242
return False
43+
# Excluding Pascal architecture, use fallback path code
44+
if target.arch in ("sm_60", "sm_61"):
45+
return False
4346
# pylint: disable=comparison-with-callable
4447
return binop == tvm.tir.generic.add and any(
4548
[

0 commit comments

Comments
 (0)