Skip to content
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.

Sign up for GitHub

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

Jump to bottom

CPU and GPU operator customization based on Taichi #518

Closed
2 tasks done
chaoming0625 opened this issue Oct 13, 2023 · 5 comments
Closed
2 tasks done

CPU and GPU operator customization based on Taichi #518

chaoming0625 opened this issue Oct 13, 2023 · 5 comments
Assignees
@Routhleck
Labels
enhancement New feature or request

Comments

@chaoming0625
Copy link
Collaborator

Please:

  • Check for duplicate requests.
  • Describe your goal, and if possible provide a code snippet with a motivating example.

See taichi-dev/taichi#6367
@chaoming0625 chaoming0625 added the enhancement New feature or request label Oct 13, 2023
@Routhleck
Copy link
Collaborator

Expose the import_cpu_memory and import_cuda_memory C API so that taichi's NdArray can directly use the memory address allocated by XLA to eliminate the memory copy consumption. taichi-dev/taichi#8368

@chaoming0625
Copy link
Collaborator Author

Does the CUDA backend also work now?

@Routhleck
Copy link
Collaborator

YES, I just tested it on CUDA backend and the results are correct!

@chaoming0625
Copy link
Collaborator Author

See #527

This was referenced Nov 8, 2023
Merged
Open
@chaoming0625
Copy link
Collaborator Author

See #553

@chaoming0625 chaoming0625 mentioned this issue Dec 29, 2023
Closed
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@Routhleck Routhleck

Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@chaoming0625 @Routhleck