38 lines
1.3 KiB
Diff
38 lines
1.3 KiB
Diff
diff --git a/warp/build_dll.py b/warp/build_dll.py
|
|
index 2218ff13..53786017 100644
|
|
--- a/warp/build_dll.py
|
|
+++ b/warp/build_dll.py
|
|
@@ -408,6 +408,7 @@ def build_dll_for_arch(args, dll_path, cpp_paths, cu_path, arch, libs: Optional[
|
|
cpp_includes += f' -I"{warp_home_path.parent}/_build/host-deps/llvm-project/release-{arch}/include"'
|
|
cuda_includes = f' -I"{cuda_home}/include"' if cu_path else ""
|
|
includes = cpp_includes + cuda_includes
|
|
+ includes += " -isystem @LIBCXX_DEV@/include/c++/v1"
|
|
|
|
if sys.platform == "darwin":
|
|
version = f"--target={arch}-apple-macos11"
|
|
@@ -441,6 +442,7 @@ def build_dll_for_arch(args, dll_path, cpp_paths, cu_path, arch, libs: Optional[
|
|
build_cmd = f'{cpp_compiler} {cpp_flags} -c "{cpp_path}" -o "{cpp_out}"'
|
|
run_cmd(build_cmd)
|
|
|
|
+ ld_inputs.append('-L"@LIBCXX_LIB@/lib" -lc++')
|
|
if cu_path:
|
|
cu_out = cu_path + ".o"
|
|
|
|
diff --git a/warp/native/crt.h b/warp/native/crt.h
|
|
index 47ef9983..89ae289b 100644
|
|
--- a/warp/native/crt.h
|
|
+++ b/warp/native/crt.h
|
|
@@ -65,6 +65,12 @@ extern "C" WP_API int _wp_isinf(double);
|
|
#include <float.h>
|
|
#include <string.h>
|
|
|
|
+#undef isfinite
|
|
+#undef isinf
|
|
+#undef isnan
|
|
+
|
|
+#include <cmath>
|
|
+
|
|
#else
|
|
|
|
// These definitions are taken from Jitify: https://github.com/NVIDIA/jitify
|