Merge pull request #1351 from rapidsai/branch-23.10

Forward-merge branch-23.10 to branch-23.12
rapidsai · Sep 26, 2023 · 19a8f8a · 19a8f8a
2 parents 19c44db + c8acf76
commit 19a8f8a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/rmm/_lib/CMakeLists.txt b/python/rmm/_lib/CMakeLists.txt
@@ -19,3 +19,5 @@ set(linked_libraries rmm::rmm)
 # Build all of the Cython targets
 rapids_cython_create_modules(SOURCE_FILES "${cython_sources}" LINKED_LIBRARIES "${linked_libraries}"
                                                                                CXX)
+# The cdef public functions in this file need to have a C ABI
+target_compile_definitions(torch_allocator PRIVATE CYTHON_EXTERN_C=extern\ "C")