DispatcherHostHIP.h

Go to the documentation of this file.

#ifndef PROTEUS_FRONTEND_DISPATCHER_HOST_HIP_H
#define PROTEUS_FRONTEND_DISPATCHER_HOST_HIP_H
 
#if PROTEUS_ENABLE_HIP
 
#include "proteus/impl/Frontend/DispatcherHost.h"
#include "proteus/impl/JitEngineDeviceHIP.h"
 
namespace proteus {
 
class DispatcherHostHIP : public DispatcherHost {
public:
  static DispatcherHostHIP &instance() {
    static DispatcherHostHIP D;
    return D;
  }
 
  StringRef getDeviceArch() const override {
    return JitEngineDeviceHIP::instance().getDeviceArch();
  }
 
private:
  // Preserve HIP cache labels when the implementation goes through host JIT.
  DispatcherHostHIP()
      : DispatcherHost("DispatcherHostHIP", JitEngineHost::instance(),
                       TargetModelType::HOST_HIP) {}
};
 
} // namespace proteus
 
#endif
 
#endif // PROTEUS_FRONTEND_DISPATCHER_HOST_HIP_H