请问op_pulgin的最终实现在哪?
lms-mt opened this issue · comments
lms-mt commented
例如,以relu为例,会调用:
at::Tensor& relu_out_npu_nocheck(at::Tensor& result, const at::Tensor& self) {
at_npu::native::OpCommand cmd;
cmd.Name("Relu")
.Input(self)
.Output(result)
.Run();
return result;
}
那么这个Relu cmd
在哪注册的?我又在哪能看到它的kernel实现呢?
yunyiyun commented
底层kernel实现暂未开放