cuda.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 cuda {
 
inline std::string CUDA_DEVICE_PREFIX = "__daisy_cuda_";
 
constexpr int CUDA_WARP_SIZE = 32;

inline data_flow::ImplementationType ImplementationType_CUDA{"CUDA"};

inline data_flow::ImplementationType ImplementationType_CUDAWithTransfers{"CUDAWithTransfers"};

inline data_flow::ImplementationType ImplementationType_CUDAWithoutTransfers{"CUDAWithoutTransfers"};
 
// Use shared GPU dimension type
using CUDADimension = gpu::GPUDimension;

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

class ScheduleType_CUDA : public gpu::ScheduleType_GPU_Base<ScheduleType_CUDA> {
public:
    static const std::string value() {
        return "CUDA";
    }
    static symbolic::Integer default_block_size_x() {
        return symbolic::integer(32);
    }
};
 
inline codegen::TargetType TargetType_CUDA_Offload{ScheduleType_CUDA_Offload::value()};
 
inline codegen::TargetType TargetType_CUDA{ScheduleType_CUDA::value()};
 
 
void cuda_error_checking(
    codegen::PrettyPrinter& stream,
    const codegen::LanguageExtension& language_extension,
    const std::string& status_variable
);
 
bool do_cuda_error_checking();
 
void check_cuda_kernel_launch_errors(
    codegen::PrettyPrinter& stream, const codegen::LanguageExtension& language_extension, bool instrumented
);
 
 
} // namespace cuda
} // namespace sdfg