at::HIPHooksInterface Struct Reference

Public Member Functions
virtual std::unique_ptr< THHState, void(*)(THHState *)>	initHIP () const
virtual std::unique_ptr< Generator >	initHIPGenerator (Context *) const
virtual bool	hasHIP () const
virtual int64_t	current_device () const
virtual Allocator *	getPinnedMemoryAllocator () const
virtual void	registerHIPTypes (Context *) const
virtual int	getNumGPUs () const

Detailed Description

Definition at line 27 of file HIPHooksInterface.h.

---

The documentation for this struct was generated from the following file:

• aten/src/ATen/detail/HIPHooksInterface.h