Tiny allow DeepGEMM on cu12.9 (#9858)
This commit is contained in:
@@ -11,9 +11,6 @@ def _compute_enable_deep_gemm():
|
|||||||
sm_version = get_device_sm()
|
sm_version = get_device_sm()
|
||||||
if sm_version < 90:
|
if sm_version < 90:
|
||||||
return False
|
return False
|
||||||
# TODO fix deepgemm cu129 fp8 issue
|
|
||||||
if torch.version.cuda == "12.9":
|
|
||||||
return False
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
import deep_gemm
|
import deep_gemm
|
||||||
|
|||||||
Reference in New Issue
Block a user