deeplink开发不能调用我写的算子

[jianlonghaha]

我在开发deleplink的时候遇到了一个问题一直不能解决，

我在/home/deeplink/deeplink.framework/dipu/third_party/DIOPI/impl里面编译好了共享库libdiopi_impl.so，so里面只实现了这么一段代码

#include <diopi/functions.h>
#include <math.h>
#include
#include "../musa_pytorch.h"
// #include "../common/common.hpp"

namespace impl {
namespace musa {
static const char* name = "MUSADevice";
DIOPI_RT_API const char* diopiGetVendorName() { return name; }
DIOPI_API diopiError_t diopiAdd(diopiContextHandle_t ctx, diopiTensorHandle_t out, diopiConstTensorHandle_t input, diopiConstTensorHandle_t other,
const diopiScalar_t* alpha) {
std::cout << "==================diopiAdd===========================\n";
return diopiSuccess;
}
编译deeplink之后在打开python之后，不能调用我写的算子，说我回退到cpu端了，如下
import torch,torch_dipu,os
dipu device will show as cuda device. if it's not expected behavior, please set env DIPU_PYTHON_DEVICE_AS_CUDA=false
Wed Jul 17 16:01:46 2024 dipu | git hash:fdc1b4c1-dirty
::diopiFill 01 is not yet implemented, fill_.Scalar will be fallback to cpu
::diopiAddScalar 01 is not yet implemented, add.Scalar_out will be fallback to cpu
::diopiAddInpScalar 01 is not yet implemented, add_.Scalar will be fallback to cpu
::diopiAddInp 01 is not yet implemented, add_.Tensor will be fallback to cpu
::diopiAdd 01 is not yet implemented, add.out will be fallback to cpu
::diopiAdd 01 is not yet implemented, add.Tensor will be fallback to cpu
这怎么解决？

[fandaoyi]

先看下 torch_dipu.so 是否 link了 diopi_impl.so 的动态库 (最新版 dipu 能保证一定 link, 老的版本有一段时间改坏了, 取决于模板编译选项). 然后看下 diopi_impl.so 里 diopiAdd 是否正确的导出了

[fandaoyi]

另外 dipu 里是否已经实现了 MUSA 相关的 接口.

[fandaoyi]

如果你的 diopi 和 dipu 是分别编译的, 编译dipu时 请确定使用 -DWITH_DIOPI_LIBRARY 指定 依赖的 diopi

[jianlonghaha]

你好，我在代码里加了extern "C" 就可以找到我写的算子了，如下，我是看有的厂商是这么做的，我也就这么写了，但是我不知道这是什么原因，哈哈，您能跟我说下是为什么吗？

namespace impl {
namespace musa {
extern "C" {
static const char* name = "MUSADevice";
DIOPI_RT_API const char* diopiGetVendorName() { return name; }
DIOPI_API diopiError_t diopiAdd(diopiContextHandle_t ctx, diopiTensorHandle_t out, diopiConstTensorHandle_t input, diopiConstTensorHandle_t other,
const diopiScalar_t* alpha) {
std::cout << "==================hello,diopiAdd===========================\n";
return diopiSuccess;
}
} //extern C
} // namespace musa
} // namespace impl

[jianlonghaha]

我可能了解了，可能因为c++函数重载的问题，不加extern "C"，C++编译器会按照C++的方式进行名称修饰
，这样因为c++的函数重载的原因会出现问题，加了extern C 函数名在链接时保持原样，这样链接器在链接阶段就不会报出未解析的外部符号错误，哈哈是这样子吗？

[fandaoyi]

diopi 对外导出的符号, 都是要加 extern C 的. 因为要求就是 C API.
不过目前 diopi 项目里的 代码大多是 有一个 adapter 封装, adapter 才是真正用来导出的, 是 c api, 而实际的实现在内部, 是 c++的