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

Warning when run tensorflow model with bladedisc #1280

Open
DY-TL opened this issue Feb 1, 2024 · 1 comment
Open

Warning when run tensorflow model with bladedisc #1280

DY-TL opened this issue Feb 1, 2024 · 1 comment

Comments

@DY-TL
Copy link

DY-TL commented Feb 1, 2024

I set the following environment variable
export TAO_ENFORCE_VERBOSE_COMPILATION_LOG=1 export TF_CPP_VMODULE=disc_compiler=1
and also got these warnings

2024-02-01 09:19:54.970256: W tao_bridge/kernels/disc_launch.cc:46] cluster: tao_cluster_2_disc_launch fallbacks to TF as tao_compiler failed.
2024-02-01 09:19:55.061115: W tao_bridge/kernels/disc_launch.cc:46] cluster: tao_cluster_2_disc_launch fallbacks to TF as tao_compiler failed.
2024-02-01 09:19:55.148396: W tao_bridge/kernels/disc_launch.cc:46] cluster: tao_cluster_2_disc_launch fallbacks to TF as tao_compiler failed.
2024-02-01 09:19:55.238986: W tao_bridge/kernels/disc_launch.cc:46] cluster: tao_cluster_2_disc_launch fallbacks to TF as tao_compiler failed.
#--# total time: 83.1417486667633s , average time:0.0831417486667633s

Any help?

@wangguofu12138
Copy link

I also face this problem, and i solve it that check cuda version >11.0, and i choose cuda version == 11.2, then make it!

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

2 participants