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 @@
#include "device.hpp"
#include "device_event.hpp"
#include "dtype.hpp"
#include "nn.hpp"
#include "ops.hpp"
#include "tensor.hpp"
......@@ -19,7 +18,6 @@ PYBIND11_MODULE(_infinicore, m) {
dtype::bind(m);
ops::bind(m);
tensor::bind(m);
pybind11_nn::bind(m);
}
} // 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