Add static_assert when sm_arch mismatches TCNN_MIN_GPU_ARCH

YOUSIKI · Feb 22, 2022 · f32361e · f32361e
1 parent ef26ff4
commit f32361e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/include/tiny-cuda-nn/common.h b/include/tiny-cuda-nn/common.h
@@ -220,6 +220,10 @@ class ScopeGuard {
 #define TCNN_HOST
 #endif
 
+#if defined(__CUDA_ARCH__)
+static_assert(__CUDA_ARCH__ >= MIN_GPU_ARCH * 10, "MIN_GPU_ARCH=" STR(TCNN_MIN_GPU_ARCH) "0 must bound __CUDA_ARCH__=" STR(__CUDA_ARCH__) " from below, but doesn't.");
+#endif
+
 template <typename T>
 TCNN_HOST_DEVICE T clamp(T val, T lower, T upper) {
 	return val < lower ? lower : (upper < val ? upper : val);