Hokori/FlashQLA-SM70-SM75
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Hokori
2026-06-15 00:30:55 +08:00
parent 3f95e2939d
commit c7eefb1166
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
flash_qla/ops/gated_delta_rule/chunk/__init__.py
View File

@@ -7,10 +7,10 @@ import tilelang
from flash_qla.utils import l2norm from flash_qla.utils import l2norm
from flash_qla.ops.utils import chunk_local_cumsum, group_reduce_vector 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 from .hopper import fused_gdr_fwd, fused_gdr_bwd, fused_gdr_h, kkt_solve
else: else:
raise ValueError("FlashQLA now support sm90 only.") raise ValueError("FlashQLA now support sm75 only.")
from .cp_context import intra_card_cp_preprocess from .cp_context import intra_card_cp_preprocess