记录 | CUDA编程中用constexpr替代__host____device__

本文主要是介绍记录 | CUDA编程中用constexpr替代__host____device__，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

比如用 __host__ & __device__ 的情况如下：

#include <cstdio>
#include <cuda_runtime.h>__host__ __device__ void say_hello(){printf("Hello, world!\n");
}__global__ void kernel(){say_hello();
}int main(){kernel<<<1, 1>>>();cudaDeviceSynchronize();say_hello();return 0;
}
}

则可以用 constexpr 来替代 __host__ __device，替代后的代码如下：

#include <cstdio>
#include <cuda_runtime.h>constexpr const char* cuthead(const char* p){return p+1;
}__global__ void kernel(){printf(cuthead("Gello, world!\n"));
}int main(){kernel<<<1, 1>>>();cudaDeviceSynchronize();printf(cuthead("Cello, world!\n"));return 0;
}

---

● 这样相当于把 constexpr 函数自动变成修饰符 __host__ __device__ ，从而两边都可以调用；
● 因为 constexpr 通常都是一些可以内联的函数，数学计算表达式之类的，一个个加上太累了，所以产生了这个需求；
● 不过必须指定 --expt-relaxed-constexpr 这个选项才能用这个特性，咱们可以用 CMake 的生成器表达式来实现只对 .cu 文件开启此选项 (不然给到 gcc 就出错了)；

# 这个.cu用nvcc编译就是这样的 
nvcc demo.cu --expt-relaxed-constexpr

● constexpr里面没办法调用 printf，也不能用 __syncthreads 之类的 GPU 特有的函数，因此也不能完全替代 __host__ 和 __device__；

这篇关于记录 | CUDA编程中用constexpr替代__host____device__的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---