Commit 1c25a902 authored by Ceng23333's avatar Ceng23333 Committed by thatPepe
Browse files

issue/688: 修复issue 634引入编译问题


Signed-off-by: default avatarCeng23333 <441651826@qq.com>
parent 2f3f4076
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
#include "device.hpp" #include "device.hpp"
#include "device_event.hpp" #include "device_event.hpp"
#include "dtype.hpp" #include "dtype.hpp"
#include "nn.hpp"
#include "ops.hpp" #include "ops.hpp"
#include "tensor.hpp" #include "tensor.hpp"
...@@ -19,7 +18,6 @@ PYBIND11_MODULE(_infinicore, m) { ...@@ -19,7 +18,6 @@ PYBIND11_MODULE(_infinicore, m) {
dtype::bind(m); dtype::bind(m);
ops::bind(m); ops::bind(m);
tensor::bind(m); tensor::bind(m);
pybind11_nn::bind(m);
} }
} // namespace infinicore } // namespace infinicore
#pragma once
#include <pybind11/pybind11.h>
#include "nn/rope.hpp"
namespace py = pybind11;
namespace infinicore::pybind11_nn {
inline void bind(py::module &m) {
bind_rope(m);
}
} // namespace infinicore::pybind11_nn
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