Unverified Commit 5e973209 authored by Zijing Liu's avatar Zijing Liu Committed by GitHub
Browse files

[BugFix] Fix type error when assign a trition kernel tensor to a torch.nn.Parameter (#28603)


Signed-off-by: default avatarZijing Liu <liuzijing2014@gmail.com>
parent e63fd445
...@@ -755,8 +755,8 @@ class Mxfp4MoEMethod(FusedMoEMethodBase): ...@@ -755,8 +755,8 @@ class Mxfp4MoEMethod(FusedMoEMethodBase):
self.w13_weight = w13_weight self.w13_weight = w13_weight
self.w2_weight = w2_weight self.w2_weight = w2_weight
layer.w13_weight = w13_weight layer.w13_weight = Parameter(w13_weight.data, requires_grad=False)
layer.w2_weight = w2_weight layer.w2_weight = Parameter(w2_weight.data, requires_grad=False)
else: else:
raise ValueError(f"Unsupported backend: {self.mxfp4_backend}") raise ValueError(f"Unsupported backend: {self.mxfp4_backend}")
......
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