From bad9a0fdc4cac06a6f0c4e2d74e6b6390d4ecfe2 Mon Sep 17 00:00:00 2001 From: youkaichao Date: Fri, 6 Sep 2024 22:17:29 -0700 Subject: [PATCH] fix typo in code --- depyf/optimization.py | 2 +- tests/test_pytorch/test_wrapper.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/depyf/optimization.py b/depyf/optimization.py index c24622a3..c9bdaf18 100644 --- a/depyf/optimization.py +++ b/depyf/optimization.py @@ -8,7 +8,7 @@ import torch -class TorchCompileWrapperWithCustomDispacther: +class TorchCompileWrapperWithCustomDispatcher: """ A wrapper class for torch.compile, with a custom dispatch logic. Subclasses should: diff --git a/tests/test_pytorch/test_wrapper.py b/tests/test_pytorch/test_wrapper.py index e8cfd115..8db3387c 100644 --- a/tests/test_pytorch/test_wrapper.py +++ b/tests/test_pytorch/test_wrapper.py @@ -2,7 +2,7 @@ import torch -from depyf.optimization import TorchCompileWrapperWithCustomDispacther +from depyf.optimization import TorchCompileWrapperWithCustomDispatcher class MyMod(torch.nn.Module): @@ -13,7 +13,7 @@ def forward(self, x: torch.Tensor, cache: Optional[torch.Tensor] = None): return x * 2 -class MyWrapper(TorchCompileWrapperWithCustomDispacther): +class MyWrapper(TorchCompileWrapperWithCustomDispatcher): def __init__(self, model): self.model = model