Unverified Commit bcf2333c authored by Matthew Bonanni's avatar Matthew Bonanni Committed by GitHub
Browse files

[CI] Fix LM Eval Large Models (H100) (#32423)


Signed-off-by: default avatarMatthew Bonanni <mbonanni@redhat.com>
parent 83239ff1
...@@ -98,7 +98,9 @@ class QuantFP8(CustomOp): ...@@ -98,7 +98,9 @@ class QuantFP8(CustomOp):
num_token_padding=self.num_token_padding, num_token_padding=self.num_token_padding,
scale_ub=scale_ub, scale_ub=scale_ub,
use_per_token_if_dynamic=self.use_per_token_if_dynamic, use_per_token_if_dynamic=self.use_per_token_if_dynamic,
group_shape=self.group_shape if self.static else None, group_shape=(self.group_shape.row, self.group_shape.col)
if self.static
else None,
) )
def forward_hip( def forward_hip(
......
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