解决了 cnplugin对算子集成是只需要.cc文件与cnplugin.h中的声明 对于.mlu中的函数需要在 头文件中声明 即类似这样,即可解决
#ifndef __HARDSWISH_KERNEL_H__
#define __HARDSWISH_KERNEL_H__
#ifdef __cplusplus
extern "C" {
#endif
void HardSwishKernel_MLU270(void * input,
void * output,
int input_size,
int data_type);
#ifdef __cplusplus
}
#endif
#endif // __HARDSWISH_KERNEL_H__
请登录后评论