Skip to content
This repository was archived by the owner on Jan 24, 2024. It is now read-only.

Nvcc compile#1416

Open
SunNy820828449 wants to merge 13 commits intoPaddlePaddle:developfrom
SunNy820828449:nvcc_compile
Open

Nvcc compile#1416
SunNy820828449 wants to merge 13 commits intoPaddlePaddle:developfrom
SunNy820828449:nvcc_compile

Conversation

@SunNy820828449
Copy link
Copy Markdown
Collaborator

using system call to use nvcc compile cuda-c cdoe and generate ptx and cubin

@paddle-bot
Copy link
Copy Markdown

paddle-bot bot commented May 11, 2023

Thanks for your contribution!

}

bool CanUseNvccCompiler() {
std::string nvcc_dir = "/usr/local/cuda/bin/nvcc";
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

这里路径是固定写死了,我们是否可以添加一个FLAGS_nvcc_cmd_path 变量,支持用户指定自己的nvcc任意路径,以满足用户想用nvcc但cinn自己找不到的灵活度?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

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

cuda一般默认都是安装再这个路径 可以加一个flag

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants