From 021ca0c31d76b1b5e6b481320bec0241acbec47b Mon Sep 17 00:00:00 2001 From: Amey Agrawal Date: Sun, 12 May 2024 01:42:41 -0700 Subject: [PATCH] minor --- simulator/profiling/mlp/mlp_impl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/simulator/profiling/mlp/mlp_impl.py b/simulator/profiling/mlp/mlp_impl.py index 24f2a09..1f8221a 100644 --- a/simulator/profiling/mlp/mlp_impl.py +++ b/simulator/profiling/mlp/mlp_impl.py @@ -4,9 +4,9 @@ # Monkey patching sarathi cuda timer to use our custom timer from simulator.profiling.cuda_timer import CudaTimer -import sarathi.metrics.cuda_timer as sarathi_cuda_timer +import sarathi.metrics.cuda_timer -sarathi_cuda_timer.CudaTimer = CudaTimer +sarathi.metrics.cuda_timer.CudaTimer = CudaTimer from sarathi.model_executor.layers.activation import SiluAndMul from sarathi.model_executor.layers.layernorm import RMSNorm