Ascend / pytorch

Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch

Home Page:https://ascend.github.io/docs/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

请问op_pulgin的最终实现在哪?

lms-mt opened this issue · comments

例如,以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实现呢?

底层kernel实现暂未开放