C++ API
torch_npu::init_npu()
torch::npu::synchronize ()
c10::npu::current_device()
Device(const std::string &device_string)
struct c10_npu::NPUEvent
class at_npu::NPUGeneratorImpl
at_npu::detail::getDefaultNPUGenerator
at_npu::detail::createNPUGenerator
class c10_npu::NPUStream
c10_npu::getNPUStreamFromPool
c10_npu::getDefaultNPUStream
c10_npu::getCurrentNPUStream
c10_npu::setCurrentNPUStream
c10_npu::getCurrentSecondaryStream
struct c10_npu::SecondaryStreamGuard
class at_npu::native::OpCommand
torch_npu::finalize_npu()
父主题:
PyTorch Adapter自定义API