Merge branch 'branch-23.10' of https://github.com/rapidsai/rmm into f…

…ea/devcontainers
rapidsai · Sep 26, 2023 · a0de1b0 · a0de1b0
2 parents 0b3ee78 + c8acf76
commit a0de1b0
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")