Unverified Commit fe543820 authored by Charlie Fu's avatar Charlie Fu Committed by GitHub
Browse files

[Rocm][Bugfix] Fix dtype not same for gemm_a4w4 op (#33734)


Signed-off-by: default avatarcharlifu <charlifu@amd.com>
parent 77c09e11
......@@ -118,7 +118,12 @@ try:
)
gemm_a4w4(
x_q, weight, x_s, weight_scale.view(x_s.dtype), y, bpreshuffle=True
x_q,
weight.view(x_q.dtype),
x_s,
weight_scale.view(x_s.dtype),
y,
bpreshuffle=True,
)
return y[:M]
else:
......
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