Custom operations for GPU using CUBLAS #19734
Unanswered
southfreebird
asked this question in
Q&A
Replies: 1 comment
-
|
For example, PyTorch has getCurrentCUDABlasHandle() function for this |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hello Team,
I'd like to create my own custom layer on C++ and CUDA using CUBLAS. What is the best way to get cublas handle in my code? Do I need to use existing pool such as this one?
Thank you!
Beta Was this translation helpful? Give feedback.
All reactions