Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

编译cuda_op时,出现error: ambiguous overload for ‘operator<<’ 以及 undefined reference to `cudnnCreateTensorDescriptor' #41

Open
KeyBor opened this issue May 10, 2024 · 1 comment

Comments

@KeyBor
Copy link

KeyBor commented May 10, 2024

环境:
系统:ubuntu20.04
cuda driver : 12.4
cudnn:8.9.7
一开始编译的时候还有关于pthread undefined reference,后面在cmakelist里面修改了cxx_flag为-lpthread,后出现:
image
image
等问题

@KeyBor
Copy link
Author

KeyBor commented May 11, 2024

cpp template <> void print_mat(__half* src, size_t size) { for (int i=0; i < size; i++) { std::cout << *((unsigned short*)(src+i)) << " "; } std::cout << std::endl; }
使用模板特化暂时解决了编译问题,但是不知道后面是否会有影响

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

No branches or pull requests

1 participant