Searched defs:MPSHooksInterface (Results 1 – 1 of 1) sorted by relevance
16 struct TORCH_API MPSHooksInterface : AcceleratorHooksInterface { struct25 virtual void initMPS() const { in initMPS()28 virtual bool hasMPS() const { in hasMPS()34 virtual const Generator& getDefaultMPSGenerator() const { in getDefaultMPSGenerator()37 virtual Allocator* getMPSDeviceAllocator() const { in getMPSDeviceAllocator()40 virtual void deviceSynchronize() const { in deviceSynchronize()43 virtual void commitStream() const { in commitStream()46 virtual void* getCommandBuffer() const { in getCommandBuffer()49 virtual void* getDispatchQueue() const { in getDispatchQueue()52 virtual void emptyCache() const { in emptyCache()[all …]