update

rusty1s · Oct 10, 2023 · 6cadf00 · 6cadf00
1 parent a165d20
commit 6cadf00
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/csrc/cpu/basis_cpu.h b/csrc/cpu/basis_cpu.h
@@ -2,6 +2,12 @@
 
 #include <torch/extension.h>
 
+#if defined(__x86_64__)
+__asm__(".symver pow,pow@GLIBC_2.2.5");
+#elif defined(__aarch64__)
+__asm__(".symver pow,pow@GLIBC_2.17");
+#endif
+
 std::tuple<torch::Tensor, torch::Tensor>
 spline_basis_fw_cpu(torch::Tensor pseudo, torch::Tensor kernel_size,
                     torch::Tensor is_open_spline, int64_t degree);