Searched defs:CUDAHooksInterface (Results 1 – 1 of 1) sorted by relevance
62 struct TORCH_API CUDAHooksInterface : AcceleratorHooksInterface { struct68 virtual void initCUDA() const { in initCUDA()76 virtual Device getDeviceFromPtr(void* /*data*/) const { in getDeviceFromPtr()80 bool isPinnedPtr(const void* data) const override { in isPinnedPtr()84 virtual bool hasCUDA() const { in hasCUDA()88 virtual bool hasCUDART() const { in hasCUDART()92 virtual bool hasMAGMA() const { in hasMAGMA()96 virtual bool hasCuDNN() const { in hasCuDNN()100 virtual bool hasCuSOLVER() const { in hasCuSOLVER()104 virtual bool hasCuBLASLt() const { in hasCuBLASLt()[all …]