File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -981,7 +981,6 @@ def forward(
981
981
# Profiling run.
982
982
return output
983
983
984
-
985
984
if attn_metadata .prefill is not None :
986
985
assert attn_metadata .num_decodes is not None and \
987
986
attn_metadata .num_prefills is not None and \
Original file line number Diff line number Diff line change @@ -91,7 +91,10 @@ def __init__(
91
91
if get_ascend_config ().use_sfa :
92
92
# Direct import instead of using try_register_lib to ensure proper error handling when
93
93
# custom_ops is necessary but not available (e.g., in DeepSeek v3.2 deployments)
94
- import custom_ops # type: ignore[import-untyped] # noqa
94
+ # yapf: disable
95
+ import custom_ops # type: ignore # noqa
96
+
97
+ # yapf: enable
95
98
logger .info (
96
99
"custom_ops module loaded successfully. Custom operators like "
97
100
"torch.ops.custom.npu_sparse_flash_attention are now available."
You can’t perform that action at this time.
0 commit comments