Skip to content

Commit

Permalink
refactor consistent module naming
Browse files Browse the repository at this point in the history
Jackmin801 committed Oct 4, 2024
1 parent 58d9b64 commit c724f2a
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions src/zeroband/C/compression.py
Original file line number Diff line number Diff line change
@@ -3,9 +3,9 @@
from torch.utils.cpp_extension import load
from pathlib import Path

COMPRESS_CSRC_PATH = Path(__file__).parent / "csrc" / "compress.cpp"
COMPRESS_CSRC_PATH = Path(__file__).parent / "csrc" / "compression.cpp"

compress_ops = load(name="compress", sources=[COMPRESS_CSRC_PATH], extra_cflags=["-O3"], verbose=False)
compress_ops = load(name="compression", sources=[COMPRESS_CSRC_PATH], extra_cflags=["-O3"], verbose=False)


def uniform_8bit_quantize(tensor: torch.Tensor, inplace: bool = True) -> Tuple[torch.Tensor, torch.Tensor]:
Original file line number Diff line number Diff line change
@@ -126,7 +126,7 @@ std::tuple<torch::Tensor, torch::Tensor> uniform_8bit_quantize(torch::Tensor ten


// PyBind11 module
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
PYBIND11_MODULE(compression, m) {
m.def(
"average_buckets",
&average_buckets_multithread,

0 comments on commit c724f2a

Please sign in to comment.