Commit 1bf9ec7f authored by Yong Tang's avatar Yong Tang Committed by Derek Murray
Browse files

Fix cmake build issues with GPU on Linux (#18775)



* Update include file for cmake

While attempting to build tensorflow with cmake for gpu the following
build error happens:
```
/home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.h:26:45: fatal error: cuda/extras/CUPTI/include/cupti.h: No such file or directory
compilation terminated.
CMakeFiles/tf_core_cpu.dir/build.make:3302: recipe for target 'CMakeFiles/tf_core_cpu.dir/home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.cc.o' failed
make[2]: *** [CMakeFiles/tf_core_cpu.dir/home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.cc.o] Error 1
CMakeFiles/Makefile2:2402: recipe for target 'CMakeFiles/tf_core_cpu.dir/all' failed
make[1]: *** [CMakeFiles/tf_core_cpu.dir/all] Error 2
Makefile:127: recipe for target 'all' failed
make: *** [all] Error 2
```
This fix tries to fix the build error above for cmake.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add xla dependency to stream_executor

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Remove nccl files temporarily

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add gomp library to libcusolver.so

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Fix cmake issue

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 4bd6ac7c
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment