Skip to content

Commit

Permalink
minor
Browse files Browse the repository at this point in the history
  • Loading branch information
Amey Agrawal authored and Amey Agrawal committed May 12, 2024
1 parent 1733968 commit 021ca0c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions simulator/profiling/mlp/mlp_impl.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 021ca0c

Please sign in to comment.