[Bugfix] Restrict Machete to only run on Hopper (#20830)

Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
Michael Goin
2025-07-13 02:34:40 +09:00
committed by GitHub
parent a86754a12b
commit 6e2c176e1f

View File

@ -32,6 +32,9 @@ class MacheteLinearKernel(MPLinearKernel):
if not current_platform.is_cuda():
return False, "Machete only supported on CUDA"
if not current_platform.is_device_capability(90):
return False, "Machete requires compute capability of 90 (Hopper)"
if c.has_g_idx and\
c.partition_weight_shape[0] != c.full_weight_shape[0]:
return False, "Act reordering currently not supported by Machete, "\