-
Notifications
You must be signed in to change notification settings - Fork 2.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[aot] Import CPU and CUDA memory for Taichi AOT #8368
Conversation
✅ Deploy Preview for docsite-preview canceled.
|
for more information, see https://pre-commit.ci
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is this file tests/python/test_ipython.ipynb
mistakenly added?
Yes, I removed
Whether |
Sorry, should have been |
Ok, I have replaced |
hmmm the easiest way is to recreate a PR without the change in |
Ok |
Issue: #
Brief Summary
🤖 Generated by Copilot at 1b3f2f9
This pull request adds new C API functions to import host or device memory pointers into TiMemory objects, which can be used by the Taichi runtime. It also adds helper functions and test cases for the new functions. The files
taichi_cpu.h
,taichi_cuda.h
,taichi_llvm_impl.cpp
,c_api_test_utils.h
,c_api_test_utils.cpp
,c_api_behavior_test.cpp
, andc_api_interop_test.cpp
are modified or created.Walkthrough
🤖 Generated by Copilot at 1b3f2f9
ti_import_cpu_memory
andti_import_cuda_memory
to the C API for importing host or device memory pointers into TiMemory objects (link, link, link)cudaMalloc
andcudaMemcpy
for allocating and copying device memory in the C API test utils (link, link)c_api_interop_test.cpp
, usingcudaMalloc
andcudaMemcpy
for the CUDA case, and verifying the data values with TiNdArray objects (link)c_api_behavior_test.cpp
to enable testing the new functions (link)@jim19930609