pyrox.dev
1diff --git a/warp/build_dll.py b/warp/build_dll.py
2index 2218ff13..53786017 100644
3--- a/warp/build_dll.py
4+++ b/warp/build_dll.py
5@@ -408,6 +408,7 @@ def build_dll_for_arch(args, dll_path, cpp_paths, cu_path, arch, libs: Optional[
6 cpp_includes += f' -I"{warp_home_path.parent}/_build/host-deps/llvm-project/release-{arch}/include"'
7 cuda_includes = f' -I"{cuda_home}/include"' if cu_path else ""
8 includes = cpp_includes + cuda_includes
9+ includes += " -isystem @LIBCXX_DEV@/include/c++/v1"
10
11 if sys.platform == "darwin":
12 version = f"--target={arch}-apple-macos11"
13@@ -441,6 +442,7 @@ def build_dll_for_arch(args, dll_path, cpp_paths, cu_path, arch, libs: Optional[
14 build_cmd = f'{cpp_compiler} {cpp_flags} -c "{cpp_path}" -o "{cpp_out}"'
15 run_cmd(build_cmd)
16
17+ ld_inputs.append('-L"@LIBCXX_LIB@/lib" -lc++')
18 if cu_path:
19 cu_out = cu_path + ".o"
20
21diff --git a/warp/native/crt.h b/warp/native/crt.h
22index 47ef9983..89ae289b 100644
23--- a/warp/native/crt.h
24+++ b/warp/native/crt.h
25@@ -65,6 +65,12 @@ extern "C" WP_API int _wp_isinf(double);
26 #include <float.h>
27 #include <string.h>
28
29+#undef isfinite
30+#undef isinf
31+#undef isnan
32+
33+#include <cmath>
34+
35 #else
36
37 // These definitions are taken from Jitify: https://github.com/NVIDIA/jitify