CUDAのArchitectureで#if分岐

目次

どうするか

nvccでコンパイルすると__CUDA_ARCH__が定義される.
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 700
// arch >=sm_70の場合のコード
#endif

注意点

cuファイル内であってもバックエンドコンパイラに渡される部分では使えません.
カテゴリー:CUDA
記事作成日:2018-08-26