Skip to content

Conversation

zhaoguochun1995
Copy link
Contributor

No description provided.

@zhaoguochun1995 zhaoguochun1995 force-pushed the zgc/deeplinx_ext_support_op_dispatch branch from 4bae507 to d0c203a Compare April 15, 2024 09:12
)


setup(
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

请教下 这个位置引入cppextension 是为了使用”TORCH_LIBRARY“吗 我记得BuildExtension和CppExtension 是在cuda 编译才用到, 如果在其他设备上用这种方式编译会不会有问题?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

setup里面的只是个草稿,还有待完善。cppextension不光光可以编译cuda kernel,也可以只编译cpp文件的。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants