Skip to content

Commit 5ed601d

Browse files
committed
flaggems black list update
1 parent 81469e4 commit 5ed601d

1 file changed

Lines changed: 75 additions & 9 deletions

File tree

vllm_fl/dispatch/config/ascend.yaml

Lines changed: 75 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -57,17 +57,16 @@ flagos_blacklist:
5757
- allclose
5858
- mean_dim
5959
- chunk_gated_delta_rule_fwd
60-
- rsqrt # performance loss
6160
- index_select
62-
- true_divide
6361
- silu
64-
- silu_and_mul # performance loss end
62+
- silu_and_mul # performance loss
6563
- to_copy
6664
- _to_copy
6765
- zeros
6866
- acos
6967
- copy_
70-
- fill_scalar_
68+
- fill_scalar_ # broken on ascend: ZeroDivisionError when filling empty tensor (grid=min(40,N)=0)
69+
- fill_tensor_ # same empty-tensor bug as fill_scalar_
7170
- sum_dim
7271
- exponential_
7372
- mm
@@ -77,17 +76,21 @@ flagos_blacklist:
7776
- floor_divide
7877
- cumsum
7978
- cumsum_out
80-
- mul
8179
- reciprocal
8280
- repeat
8381
- randn
84-
- add
8582
- ge_scalar
86-
- sub
83+
- pow_scalar # broken on ascend: tl_extra_shim.pow is a libdevice stub returning None
84+
- pow_tensor_scalar # same broken shim pow
85+
- pow_tensor_scalar_ # same broken shim pow
86+
- pow_tensor_tensor # same broken shim pow
87+
- pow_tensor_tensor_ # same broken shim pow
88+
- gelu # broken on ascend: gelu_tanh kernel calls the stubbed shim pow
89+
- gelu_ # same broken shim pow
90+
- gelu_backward # same broken shim pow
8791
- bitwise_and
8892
- bitwise_not
8993
- slice_scatter
90-
- fill_tensor_
9194
- conv1d
9295
- conv2d
9396
- uniform_
@@ -107,8 +110,71 @@ flagos_blacklist:
107110
- _flash_attention_forward
108111
- scatter
109112
- scatter_
113+
# top5 kernel flaggems(1>%)
114+
- addmm
115+
- argmax
116+
- add
117+
- mul
118+
- index_put_
119+
# torch_npu's index_put_ internally dispatches to aten::_index_put_impl_, which
120+
# FlagGems also registers -- blacklisting index_put_ alone is NOT enough.
121+
# Its triton kernel launches with grid=0 on empty indices (coreDim=0 -> crash).
122+
- _index_put_impl_
123+
124+
# --- Added from full-op sweep (scripts/flaggems_sweep.py, 2026-08-07) ---
125+
# Triton compile failures on ascend (stubbed libdevice shims etc.)
126+
- tan
127+
- tan_
128+
- tanh
129+
- tanh_
130+
- atan
131+
- atan_
132+
- atan2
133+
- atan2_out
134+
- sigmoid
135+
- sigmoid_
136+
- glu
137+
- nan_to_num
138+
- isfinite
139+
- isinf
140+
- isnan
141+
- floor_divide_
142+
- silu_backward
143+
- nll_loss_backward
144+
- batch_norm
145+
- weight_norm_interface_backward
146+
# MLIR lowering failures (triton-ascend adapter)
147+
- var
148+
- var_correction
149+
- var_dim
150+
- softmax_backward
151+
# FlagGems impls that hard-require CUDA tensors
152+
- arcsinh
153+
- arcsinh_out
154+
- i0_
155+
- prelu
156+
- special_i0e
157+
- special_i0e_out
158+
- special_i1
159+
- select_backward
160+
- soft_margin_loss
161+
- _safe_softmax
162+
- t_copy
163+
- t_copy_out
164+
- lift_fresh_copy
165+
- zero
166+
- zero_out
167+
# Real limitation: FlagGems impl raises "varlen is not supported yet".
168+
# (the yaml entry "_flash_attention_forward" above never matched this func name)
169+
- flash_attention_forward
170+
# constant_pad_nd hangs the aicore in the ViT attention pad path
171+
# (head_size 72->128 F.pad on large dummy images), found by bisect 2026-08-13.
172+
- pad
173+
# aten::constant_pad_nd is bound to func 'constant_pad_nd', NOT 'pad' --
174+
# blacklisting 'pad' alone does NOT disable it (F.pad constant mode dispatches here).
175+
- constant_pad_nd
110176

111177
# OOT (Out-of-Tree) operator blacklist
112178
# These operators will NOT be registered as OOT replacements.
113179
# oot_blacklist:
114-
# - fused_moe
180+
# - fused_moe

0 commit comments

Comments
 (0)