Searched defs:HIPHooksInterface (Results 1 – 1 of 1) sorted by relevance
24 struct TORCH_API HIPHooksInterface : AcceleratorHooksInterface { struct30 virtual void initHIP() const { in initHIP()34 virtual std::unique_ptr<c10::GeneratorImpl> initHIPGenerator(Context*) const { in initHIPGenerator()38 virtual bool hasHIP() const { in hasHIP()42 virtual c10::DeviceIndex current_device() const { in current_device()46 bool isPinnedPtr(const void* data) const override { in isPinnedPtr()71 TORCH_DECLARE_REGISTRY(HIPHooksRegistry, HIPHooksInterface, HIPHooksArgs); argument