mirror/tensorflow
1
0
Fork 0
mirror of https://github.com/tensorflow/tensorflow.git synced 2024-11-21 21:05:19 +00:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Use if macros

Browse Source
This commit is contained in:
Kaixi Hou 2021-03-16 17:41:34 -07:00
parent 7e8898c42f
commit 5ffdb1f896
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/stream_executor/stream.h
View File

@ -29,7 +29,6 @@ limitations under the License.
#include "tensorflow/core/platform/macros.h"
#include "tensorflow/core/platform/thread_annotations.h"
#include "tensorflow/stream_executor/blas.h"
#include "tensorflow/stream_executor/cuda/cuda_dnn.h"
#include "tensorflow/stream_executor/device_memory.h"
#include "tensorflow/stream_executor/dnn.h"
#include "tensorflow/stream_executor/event.h"
@ -42,6 +41,10 @@ limitations under the License.
#include "tensorflow/stream_executor/stream_executor_pimpl.h"
#include "tensorflow/stream_executor/temporary_memory_manager.h"
#if GOOGLE_CUDA
#include "tensorflow/stream_executor/cuda/cuda_dnn.h"
#endif // GOOGLE_CUDA
namespace stream_executor {
namespace host {
@ -363,6 +366,7 @@ class Stream {
DeviceMemory<OutputType> *output, ScratchAllocator *scratch_allocator,
const dnn::AlgorithmConfig &plan_config,
dnn::ProfileResult *output_profile_result) {
#if GOOGLE_CUDA
dnn::DnnSupport *dnn = parent_->AsDnn();
if (dnn) {
gpu::CudnnSupport *cudnn_dnn = dynamic_cast<gpu::CudnnSupport*>(dnn);
@ -373,6 +377,8 @@ class Stream {
output_descriptor, *output, convolution_descriptor, plan_config,
scratch_allocator, output_profile_result);
}
#endif // GOOGLE_CUDA
return port::UnimplementedError("DNN library is not found.");
}
port::Status FusedConvolveWithAlgorithm(