Fix compiler warnings.

k2-fsa · Oct 12, 2021 · 4066bd0 · 4066bd0
1 parent 68a0753
commit 4066bd0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/k2/csrc/cub.h b/k2/csrc/cub.h
@@ -30,6 +30,14 @@
 // that k2 and PyTorch use a different copy
 // of CUB.
 
+#ifdef CUB_NS_PREFIX
+#undef CUB_NS_PREFIX
+#endif
+
+#ifdef CUB_NS_POSTFIX
+#undef CUB_NS_POSTFIX
+#endif
+
 #define CUB_NS_PREFIX namespace k2 {
 #define CUB_NS_POSTFIX }