We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 38e50ed + 09ee3d6 commit 2e757bbCopy full SHA for 2e757bb
1 file changed
src/rtc_kernel.cu
@@ -132,6 +132,7 @@ CudaRtcKernel::CudaRtcKernel(const std::string& name, const std::string& kernel_
132
133
std::vector<std::string> opts = {
134
fmt::format("--gpu-architecture=compute_{}", cc),
135
+ fmt::format("-DTCNN_HALF_PRECISION={}", TCNN_HALF_PRECISION),
136
fmt::format("-DTCNN_MIN_GPU_ARCH={}", cc),
137
"--std=c++14",
138
#ifdef TCNN_RTC_USE_FAST_MATH
0 commit comments