[Pytorch CUDA] 如何使用xmake为Pytorch编译CUDA算子？

[shenmishajing]

你在什么场景下需要该功能？

已经有很多教程关于使用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

[Issues-translate-bot]

Bot detected the issue body's language is not English, translate it automatically.

---

Title: [Pytorch CUDA] How to use xmake to compile CUDA operators for Pytorch?

[waruqi]

没试过，你可以研究下

[Issues-translate-bot]

Bot detected the issue body's language is not English, translate it automatically.

---

Haven't tried it, you can research it

[waruqi]

这个应该就是 用 libtorch 包 + cuda 编译，好像也没啥特别的。。

add_requires("libtorch")
target("test")
    set_kind("binary")
    add_files("src/*.cu")
    add_files("src/*.cpp")
    add_packages("libtorch")

[shenmishajing]

能用conda环境里安装的Pytorch吗？cmake那边是在conda环境里安装一个Pytorch然后把相关的路径添加到CMAKE_PREFIX_PATH里就可以了，这样就不用再单独安装一遍libtorch了。

[wangyexiang]

请问，你搞定了吗？@shenmishajing