diff --git a/src/rtc_kernel.cu b/src/rtc_kernel.cu index 9794304f..b7f0ab8b 100644 --- a/src/rtc_kernel.cu +++ b/src/rtc_kernel.cu @@ -132,6 +132,7 @@ CudaRtcKernel::CudaRtcKernel(const std::string& name, const std::string& kernel_ std::vector opts = { fmt::format("--gpu-architecture=compute_{}", cc), + fmt::format("-DTCNN_HALF_PRECISION={}", TCNN_HALF_PRECISION), fmt::format("-DTCNN_MIN_GPU_ARCH={}", cc), "--std=c++14", #ifdef TCNN_RTC_USE_FAST_MATH