Searched defs:MTIAHooksInterface (Results 1 – 1 of 1) sorted by relevance
26 struct TORCH_API MTIAHooksInterface : AcceleratorHooksInterface { struct34 virtual void initMTIA() const { in initMTIA()41 virtual bool hasMTIA() const { in hasMTIA()45 DeviceIndex deviceCount() const override { in deviceCount()49 virtual void deviceSynchronize(c10::DeviceIndex device_index) const { in deviceSynchronize()53 virtual std::string showConfig() const { in showConfig()57 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext()61 void setCurrentDevice(DeviceIndex device) const override { in setCurrentDevice()65 DeviceIndex getCurrentDevice() const override { in getCurrentDevice()70 DeviceIndex exchangeDevice(DeviceIndex device) const override { in exchangeDevice()[all …]