Searched defs:AcceleratorHooksInterface (Results 1 – 1 of 1) sorted by relevance
14 struct TORCH_API AcceleratorHooksInterface { struct22 virtual DeviceIndex deviceCount() const { in deviceCount()26 virtual void setCurrentDevice(DeviceIndex device) const { in setCurrentDevice()30 virtual DeviceIndex getCurrentDevice() const { in getCurrentDevice()35 virtual DeviceIndex exchangeDevice(DeviceIndex device) const { in exchangeDevice()40 virtual DeviceIndex maybeExchangeDevice(DeviceIndex device) const { in maybeExchangeDevice()45 virtual bool isPinnedPtr(const void* data) const { in isPinnedPtr()49 virtual Allocator* getPinnedMemoryAllocator() const { in getPinnedMemoryAllocator()