Searched defs:XPUHooksInterface (Results 1 – 1 of 1) sorted by relevance
14 struct TORCH_API XPUHooksInterface : AcceleratorHooksInterface{ struct17 virtual void initXPU() const { in initXPU()23 virtual bool hasXPU() const { in hasXPU()27 virtual std::string showConfig() const { in showConfig()33 virtual int32_t getGlobalIdxFromDevice(const Device& device) const { in getGlobalIdxFromDevice()45 virtual DeviceIndex getNumGPUs() const { in getNumGPUs()49 virtual DeviceIndex current_device() const { in current_device()76 C10_DECLARE_REGISTRY(XPUHooksRegistry, XPUHooksInterface, XPUHooksArgs); argument