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: ...@@ -118,7 +118,12 @@ try:
) )
gemm_a4w4( 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] return y[:M]
else: 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