[Pytorch CUDA] 如何使用xmake为Pytorch编译CUDA算子? #3962
Unanswered
shenmishajing
asked this question in
Q&A
Replies: 4 comments 2 replies
-
Title: [Pytorch CUDA] How to use xmake to compile CUDA operators for Pytorch? |
Beta Was this translation helpful? Give feedback.
0 replies
-
没试过,你可以研究下 |
Beta Was this translation helpful? Give feedback.
0 replies
-
Haven't tried it, you can research it |
Beta Was this translation helpful? Give feedback.
0 replies
-
这个应该就是 用 libtorch 包 + cuda 编译,好像也没啥特别的。。 add_requires("libtorch")
target("test")
set_kind("binary")
add_files("src/*.cu")
add_files("src/*.cpp")
add_packages("libtorch") |
Beta Was this translation helpful? Give feedback.
2 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
你在什么场景下需要该功能?
已经有很多教程关于使用cmake为Pytorch编译C++或CUDA算子了,xmake可以实现相同的东西吗?
一个可能的CMakeLists.txt:
https://github.com/godweiyang/NN-CUDA-Example/blob/master/pytorch/CMakeLists.txt
描述可能的解决方案
一个能编译类似的Pytorch C++/CUDA 算子的xmake.lua,或者有可能的话,一个CMakeLists.txt转xmake.lua的工具?
描述你认为的候选方案
No response
其他信息
No response
Beta Was this translation helpful? Give feedback.
All reactions