Skip to content

Commit 515f764

Browse files
author
cloudforge1
committed
fix: add get_padding_offset to GPU ops stub
layers/utils.py imports get_padding_offset at module level when CUDA is available. This caused collection error in CI.
1 parent cda64ef commit 515f764

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/layers/test_fused_moe_deepgemm_backend.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
"MoeWna16MarlinGemmApi",
3737
"tritonmoe_preprocess_func",
3838
"gptq_marlin_repack",
39+
"get_padding_offset",
3940
):
4041
if not hasattr(_gpu, _a):
4142
setattr(_gpu, _a, None)

0 commit comments

Comments
 (0)