Skip to content

GitLab

"...git@developer.sourcefind.cn:2222/OpenDAS/vllm_cscc.git" did not exist on "a88b3be7c4c494c896f4a88ca8e6bfc1083625e0"
Commit 2ced1fb2 authored by zhuwenwen's avatar zhuwenwen
Browse files

update fix_functionalization.py

parent 3e485650
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/compilation/fix_functionalization.py
View file @ 2ced1fb2
...@@ -67,7 +67,7 @@ class FixFunctionalizationPass(VllmInductorPass): ...@@ -67,7 +67,7 @@ class FixFunctionalizationPass(VllmInductorPass):
# self.defunctionalize(graph, node, mutated_args) # self.defunctionalize(graph, node, mutated_args)
# elif at_target == torch.ops._C.rms_norm_dynamic_per_token_quant.default: # noqa: E501 # elif at_target == torch.ops._C.rms_norm_dynamic_per_token_quant.default: # noqa: E501
# mutated_args = {1: 'result', 2: 'scale', 3: 'residual'} # mutated_args = {1: 'result', 2: 'scale', 3: 'residual'}
self.defunctionalize(graph, node, mutated_args) # self.defunctionalize(graph, node, mutated_args)
elif at_target in [ elif at_target in [
torch.ops._C.rms_norm.default, torch.ops._C.rms_norm.default,
# torch.ops._C.rms_norm_static_fp8_quant.default, # torch.ops._C.rms_norm_static_fp8_quant.default,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment