feat(cmake): expose TCNN_HALF_PRECISION to parent

Author: Tom94

CMakeLists.txt

@@ -1,5 +1,5 @@
 # Copyright (c) 2020-2025, NVIDIA CORPORATION.  All rights reserved.
-#
+# 
 # Redistribution and use in source and binary forms, with or without modification, are permitted
 # provided that the following conditions are met:
 #     * Redistributions of source code must retain the above copyright notice, this list of
@@ -10,7 +10,7 @@
 #     * Neither the name of the NVIDIA CORPORATION nor the names of its contributors may be used
 #       to endorse or promote products derived from this software without specific prior written
 #       permission.
-#
+# 
 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
 # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 # FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE
@@ -221,6 +221,7 @@ message(STATUS "Targeting CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}")
 if (TCNN_HAS_PARENT)
 	set(TCNN_CUDA_ARCHITECTURES ${CMAKE_CUDA_ARCHITECTURES} PARENT_SCOPE)
 	set(TCNN_CUDA_VERSION ${CUDA_VERSION} PARENT_SCOPE)
+	set(TCNN_HALF_PRECISION ${TCNN_HALF_PRECISION} PARENT_SCOPE)
 endif()
 
 if (MIN_GPU_ARCH LESS_EQUAL 70)

dependencies/cuda-cmake-github-actions/scripts/actions/install_cuda_windows.ps1

@@ -55,7 +55,6 @@ $CUDA_PACKAGES_IN = @(
 	"cudart";
 )
 
-
 ## -------------------
 ## Select CUDA version
 ## -------------------

include/tiny-cuda-nn/gpu_memory.h

@@ -695,6 +695,7 @@ inline GPUMemoryArena::Allocation allocate_workspace(cudaStream_t stream, size_t
 	if (!arena) {
 		arena = std::make_shared<GPUMemoryArena>();
 	}
+
 	return GPUMemoryArena::Allocation{stream, arena->allocate(n_bytes), arena};
 }
 

scripts/common.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 
 # Copyright (c) 2020-2025, NVIDIA CORPORATION.  All rights reserved.
-#
+# 
 # Redistribution and use in source and binary forms, with or without modification, are permitted
 # provided that the following conditions are met:
 #     * Redistributions of source code must retain the above copyright notice, this list of
@@ -12,7 +12,7 @@
 #     * Neither the name of the NVIDIA CORPORATION nor the names of its contributors may be used
 #       to endorse or promote products derived from this software without specific prior written
 #       permission.
-#
+# 
 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
 # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 # FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE

src/rtc_kernel.cu

@@ -122,7 +122,7 @@ CudaRtcKernel::CudaRtcKernel(const std::string& name, const std::string& kernel_
 
 	ScopeGuard cleanup_guard{[&]() { clear(); }};
 
-	auto start_time = std::chrono::steady_clock::now();
+	const auto start_time = std::chrono::steady_clock::now();
 
 	uint32_t cc = cuda_supported_compute_capability();