From 9f9179e052083ab863a5daaa3c0f7669a43fa7d6 Mon Sep 17 00:00:00 2001 From: dushuaishuai Date: Fri, 13 Dec 2024 16:25:23 +0800 Subject: [PATCH] Fix missing memory_efficient params --- workload_generator/mocked_model/AiobMegatron.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/workload_generator/mocked_model/AiobMegatron.py b/workload_generator/mocked_model/AiobMegatron.py index 60e10d3..599107d 100755 --- a/workload_generator/mocked_model/AiobMegatron.py +++ b/workload_generator/mocked_model/AiobMegatron.py @@ -361,7 +361,7 @@ def __init__(self, args=None): @cuda_timing_decorator def _apply(self, hidden_states): output_lay = FastLayerNormFN.apply( - hidden_states, self.lay_weight, self.bias, 1e-05 + hidden_states, self.lay_weight, self.bias, 1e-05, 1 ) return output_lay