fuck 根本没改算力判断，我真是操了

2026-06-15 00:30:55 +08:00
parent 3f95e2939d
commit c7eefb1166
1 changed files with 2 additions and 2 deletions
--- a/flash_qla/ops/gated_delta_rule/chunk/init.py
+++ b/flash_qla/ops/gated_delta_rule/chunk/init.py
@@ -7,10 +7,10 @@ import tilelang
 from flash_qla.utils import l2norm
 from flash_qla.ops.utils import chunk_local_cumsum, group_reduce_vector

-if tilelang.contrib.nvcc.get_target_compute_version() == "9.0":
+if tilelang.contrib.nvcc.get_target_compute_version() == "7.5":
    from .hopper import fused_gdr_fwd, fused_gdr_bwd, fused_gdr_h, kkt_solve
 else:
-    raise ValueError("FlashQLA now support sm90 only.")
+    raise ValueError("FlashQLA now support sm75 only.")
 from .cp_context import intra_card_cp_preprocess