Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
f2d16207
Unverified
Commit
f2d16207
authored
Mar 26, 2026
by
Andreas Karatzas
Committed by
GitHub
Mar 26, 2026
Browse files
[ROCm][CI] Fix flaky GPTQ compile correctness test (#38161)
Signed-off-by:
Andreas Karatzas
<
akaratza@amd.com
>
parent
37a83007
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
41 additions
and
34 deletions
+41
-34
tests/compile/fullgraph/test_basic_correctness.py
tests/compile/fullgraph/test_basic_correctness.py
+2
-2
vllm/model_executor/model_loader/weight_utils.py
vllm/model_executor/model_loader/weight_utils.py
+39
-32
No files found.
tests/compile/fullgraph/test_basic_correctness.py
View file @
f2d16207
...
...
@@ -137,6 +137,7 @@ def test_compile_correctness(
all_args
.
append
(
final_args
+
[
f
"-cc.mode=
{
mode
.
name
}
"
,
"-cc.backend=inductor"
]
)
all_envs
.
append
({})
# inductor will change the output, so we only compare if the output
# is close, not exactly the same.
...
...
@@ -157,6 +158,5 @@ def test_compile_correctness(
]:
all_args
.
append
(
final_args
+
[
f
"-cc.mode=
{
mode
.
name
}
"
,
"-cc.backend=eager"
])
all_envs
.
append
({})
all_envs
.
append
({})
compare_all_settings
(
model
,
all_args
*
3
,
all_envs
,
method
=
method
)
compare_all_settings
(
model
,
all_args
,
all_envs
,
method
=
method
)
vllm/model_executor/model_loader/weight_utils.py
View file @
f2d16207
...
...
@@ -1348,7 +1348,14 @@ def initialize_single_dummy_weight(
high
:
float
=
1e-3
,
seed
:
int
=
1234
,
)
->
None
:
if
torch
.
is_floating_point
(
param
):
if
not
torch
.
is_floating_point
(
param
):
if
current_platform
.
is_rocm
():
# On ROCm, integer params (e.g. GPTQ qweight/qzeros) are left
# as torch.empty() by default, giving non-deterministic values
# across processes. Zero them for reproducibility.
param
.
zero_
()
return
if
current_platform
.
is_tpu
():
generator
=
torch
.
Generator
(
device
=
"cpu"
)
generator
.
manual_seed
(
seed
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment