Searched refs:CUDAHooksInterface (Results 1 – 5 of 5) sorted by relevance
25 static CUDAHooksInterface* cuda_hooks = nullptr;27 const CUDAHooksInterface& getCUDAHooks() { in getCUDAHooks()41 cuda_hooks = new CUDAHooksInterface(); in getCUDAHooks()46 cuda_hooks = new CUDAHooksInterface(); in getCUDAHooks()53 C10_DEFINE_REGISTRY(CUDAHooksRegistry, CUDAHooksInterface, CUDAHooksArgs)
62 struct TORCH_API CUDAHooksInterface : AcceleratorHooksInterface { struct65 ~CUDAHooksInterface() override = default;204 TORCH_DECLARE_REGISTRY(CUDAHooksRegistry, CUDAHooksInterface, CUDAHooksArgs); argument209 TORCH_API const CUDAHooksInterface& getCUDAHooks();
20 struct CUDAHooks : public at::CUDAHooksInterface {
971 "aten/src/ATen/detail/CUDAHooksInterface.cpp",988 "aten/src/ATen/detail/CUDAHooksInterface.h",
2076 "aten/src/ATen/detail/CUDAHooksInterface.cpp",