Skip to content

Commit 344e42b

Browse files
cyber-pioneerphysics31415926Copilot
authored
Update gems blacklist for nvidia (#222)
### PR Category Core ### PR Type Improvements ### Description Update gems blacklist for nvidia platform --------- Co-authored-by: Physics <3150105638@zju.edu.cn> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
1 parent b6cd46e commit 344e42b

2 files changed

Lines changed: 1 addition & 6 deletions

File tree

vllm_fl/dispatch/backends/flaggems/flaggems.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@ def attention_backend(self, use_mla: bool = False, use_sparse: bool = False) ->
158158

159159
if use_sparse:
160160
raise ValueError("use_sparse=True requires use_mla=True.")
161+
# TODO: return "vllm_fl.dispatch.backends.flaggems.impl.attention.AttentionFLBackend"
161162

162163
return AttentionBackendEnum.TRITON_ATTN.get_path()
163164

vllm_fl/dispatch/config/nvidia.yaml

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,6 @@ op_backends:
4545

4646
# FlagOS operator blacklist
4747
flagos_blacklist:
48-
- copy_
49-
- to_copy
50-
- cat
51-
- index
52-
- mul
53-
- pow_scalar
5448
- index_put_
5549
- index_put
5650
- _index_put_impl_

0 commit comments

Comments
 (0)