Unverified Commit a5772f67 authored by qwopqwop200's avatar qwopqwop200 Committed by GitHub
Browse files

Rename pybind.cpp to pybind_linux.cpp

parent d76125bf
......@@ -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("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);
}
\ 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