//===--- Cuda.h - Utilities for compiling CUDA code ------------*- C -*-===//

//

// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.

// See https://llvm.org/LICENSE.txt for license information.

// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

//

//===----------------------------------------------------------------------===//

#ifndef LLVM_CLANG_BASIC_CUDA_H

#define LLVM_CLANG_BASIC_CUDA_H

namespace llvm {

class StringRef;

class Twine;

class VersionTuple;

} // namespace llvm

namespace clang {

enum class CudaVersion {

};

const char *CudaVersionToString(CudaVersion V);

// Input is "Major.Minor"

CudaVersion CudaStringToVersion(const llvm::Twine &S);

enum class CudaArch {

};

static inline bool IsNVIDIAGpuArch(CudaArch A) {

return A >= CudaArch::SM_20 && A < CudaArch::GFX600;

}

static inline bool IsAMDGpuArch(CudaArch A) {

return A >= CudaArch::GFX600 && A < CudaArch::LAST;

}

const char *CudaArchToString(CudaArch A);

const char *CudaArchToVirtualArchString(CudaArch A);

// The input should have the form "sm_20".

CudaArch StringToCudaArch(llvm::StringRef S);

/// Get the earliest CudaVersion that supports the given CudaArch.

CudaVersion MinVersionForCudaArch(CudaArch A);

/// Get the latest CudaVersion that supports the given CudaArch.

CudaVersion MaxVersionForCudaArch(CudaArch A);

// Various SDK-dependent features that affect CUDA compilation

enum class CudaFeature {

};

CudaVersion ToCudaVersion(llvm::VersionTuple);

bool CudaFeatureEnabled(llvm::VersionTuple, CudaFeature);

bool CudaFeatureEnabled(CudaVersion, CudaFeature);

} // namespace clang

#endif