C++ API
概述
(beta)torch_npu::init_npu
(beta)torch_npu::finalize_npu
(beta)torch::npu::synchronize
(beta)c10::npu::current_device
(beta)at::Device
(beta)struct c10_npu::NPUEvent
(beta)class at_npu::NPUGeneratorImpl
(beta)at_npu::detail::getDefaultNPUGenerator
(beta)at_npu::detail::createNPUGenerator
(beta)class c10_npu::NPUStream
(beta)c10_npu::getNPUStreamFromPool
(beta)c10_npu::getDefaultNPUStream
(beta)c10_npu::getCurrentNPUStream
(beta)c10_npu::setCurrentNPUStream
(beta)class at_npu::native::OpCommand
(beta)struct c10_npu::NPUHooksInterface
(beta)struct c10_npu::NPUHooksArgs
(beta)c10_npu::device_count
(beta)c10_npu::GetDevice
(beta)c10_npu::SetDevice
(beta)c10_npu::current_device
(beta)c10_npu::set_device
(beta)c10_npu::warning_state
(beta)c10_npu::warn_or_error_on_sync
(beta)at_npu::native::get_npu_format
(beta)at_npu::native::get_npu_storage_sizes
(beta)at_npu::native::npu_format_cast
(beta)at_npu::native::empty_with_format
(beta)c10_npu::c10_npu_get_error_message
(beta)at_npu::native::npu_dropout_gen_mask
父主题:
Ascend Extension for PyTorch自定义API