kernels-community
/

rotary

Model card Files Files and versions

danieldk HF Staff commited on 3 days ago

Commit

c4d3c95

·

1 Parent(s): e94ff91

Fix CUDA build

Files changed (1) hide show

rotary/rotary_cuda.cu +1 -1

rotary/rotary_cuda.cu CHANGED Viewed

@@ -6,7 +6,7 @@
 #include <ATen/native/TensorIterator.h>
 #include <ATen/native/cuda/Loops.cuh>
-void apply_rotary_cuda(torch::Tensor const &x1, torch::Tensor const &x2,
                        torch::Tensor const &cos, torch::Tensor const &sin,
                        torch::Tensor &out1, torch::Tensor &out2,
                        bool const conj) {

 #include <ATen/native/TensorIterator.h>
 #include <ATen/native/cuda/Loops.cuh>
+void _apply_rotary(torch::Tensor const &x1, torch::Tensor const &x2,
                        torch::Tensor const &cos, torch::Tensor const &sin,
                        torch::Tensor &out1, torch::Tensor &out2,
                        bool const conj) {