"tests/test_device/test_ipu/test_ipu_dataloder.py" did not exist on "df4658206dd71723104636e082202fd778ef1bc6"
Unverified Commit 9bd4872a authored by Trevor Morris's avatar Trevor Morris Committed by GitHub
Browse files

[bugfix] Fix typo in modelopt quant: 'FusedMoE' object has no attribute 'local_num_experts' (#8768)

parent 2fa0462c
......@@ -776,7 +776,7 @@ class ModelOptNvFp4FusedMoEMethod(FusedMoEMethodBase):
# GEMM 1
w13_weight = ModelWeightParameter(
data=torch.empty(
layer.local_num_experts,
layer.num_local_experts,
2 * intermediate_size_per_partition,
# 2 fp4 items are packed in the input dimension
hidden_size // 2,
......
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