Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "f367d621c629eade67347bb4a0b9cb1ebe261641"
Unverified Commit a5772f67 authored by qwopqwop200's avatar qwopqwop200 Committed by GitHub
Browse files

Rename pybind.cpp to pybind_linux.cpp

parent d76125bf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
awq_cuda/pybind.cpp → awq_cuda/pybind_linux.cpp
View file @ a5772f67
...@@ -16,4 +16,4 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) ...@@ -16,4 +16,4 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
py::arg("q"), py::arg("k"), py::arg("v"), py::arg("k_cache"), py::arg("v_cache"), py::arg("q"), py::arg("k"), py::arg("v"), py::arg("k_cache"), py::arg("v_cache"),
py::arg("length_per_sample_"), py::arg("alibi_slopes_"), py::arg("timestep"), py::arg("rotary_embedding_dim")=0, py::arg("length_per_sample_"), py::arg("alibi_slopes_"), py::arg("timestep"), py::arg("rotary_embedding_dim")=0,
py::arg("rotary_base")=10000.0f, py::arg("neox_rotary_style")=true); py::arg("rotary_base")=10000.0f, py::arg("neox_rotary_style")=true);
} }
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment