rocm.h

Go to the documentation of this file.

#pragma once
 
#include <string>
 
#include "sdfg/codegen/instrumentation/instrumentation_info.h"
#include "sdfg/codegen/language_extension.h"
#include "sdfg/codegen/utils.h"
#include "sdfg/targets/gpu/gpu_offload_schedule_type.h"
#include "sdfg/targets/gpu/gpu_schedule_type.h"
#include "sdfg/targets/gpu/gpu_types.h"
 
namespace sdfg {
namespace rocm {
 
inline std::string ROCM_DEVICE_PREFIX = "__daisy_hip_";
 
constexpr int ROCM_WARP_SIZE = 64;

inline data_flow::ImplementationType ImplementationType_ROCM{"ROCM"};

inline data_flow::ImplementationType ImplementationType_ROCMWithTransfers{"ROCMWithTransfers"};

inline data_flow::ImplementationType ImplementationType_ROCMWithoutTransfers{"ROCMWithoutTransfers"};
 
// Use shared GPU dimension type
using ROCMDimension = gpu::GPUDimension;

class ScheduleType_ROCM_Offload : public gpu::ScheduleType_GPU_Offload {
public:
    static const std::string value() {
        return "ROCM_Offload";
    }
};

class ScheduleType_ROCM : public gpu::ScheduleType_GPU_Base<ScheduleType_ROCM> {
public:
    static const std::string value() {
        return "ROCM";
    }
    static symbolic::Integer default_block_size_x() {
        return symbolic::integer(64);
    }
};
 
inline codegen::TargetType TargetType_ROCM_Offload{ScheduleType_ROCM_Offload::value()};
 
inline codegen::TargetType TargetType_ROCM{ScheduleType_ROCM::value()};
 
 
void rocm_error_checking(
    codegen::PrettyPrinter& stream,
    const codegen::LanguageExtension& language_extension,
    const std::string& status_variable
);
 
bool do_rocm_error_checking();

int rocm_wavefront_size();
 
void check_rocm_kernel_launch_errors(codegen::PrettyPrinter& stream, const codegen::LanguageExtension& language_extension);
 
} // namespace rocm
} // namespace sdfg