Commit fec7bee5 authored by Chris Sullivan's avatar Chris Sullivan Committed by Robert Kimball

Modified CUDNN headers throughout nGraph: cudnn_v7.h -> cudnn.h. (#879)

This better supports non-ubuntu/debian systems.
parent 1a73f10c
...@@ -22,7 +22,7 @@ ...@@ -22,7 +22,7 @@
#include <cublas_v2.h> #include <cublas_v2.h>
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cudnn_v7.h> #include <cudnn.h>
#include "ngraph/axis_set.hpp" #include "ngraph/axis_set.hpp"
#include "ngraph/runtime/gpu/gpu_runtime_context.hpp" #include "ngraph/runtime/gpu/gpu_runtime_context.hpp"
......
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
#include <cublas_v2.h> #include <cublas_v2.h>
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cudnn_v7.h> #include <cudnn.h>
#include <iostream> #include <iostream>
#include <nvrtc.h> #include <nvrtc.h>
#include <string> #include <string>
......
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,7 @@
#include <cublas_v2.h> #include <cublas_v2.h>
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cudnn_v7.h> #include <cudnn.h>
#include <fstream> #include <fstream>
#include <memory> #include <memory>
#include <string> #include <string>
...@@ -287,7 +287,7 @@ void runtime::gpu::GPU_ExternalFunction::compile() ...@@ -287,7 +287,7 @@ void runtime::gpu::GPU_ExternalFunction::compile()
#include <cublas_v2.h> #include <cublas_v2.h>
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cudnn_v7.h> #include <cudnn.h>
#include "ngraph/descriptor/input.hpp" #include "ngraph/descriptor/input.hpp"
#include "ngraph/descriptor/layout/dense_tensor_view_layout.hpp" #include "ngraph/descriptor/layout/dense_tensor_view_layout.hpp"
......
...@@ -24,7 +24,7 @@ ...@@ -24,7 +24,7 @@
#include <cublas_v2.h> #include <cublas_v2.h>
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cudnn_v7.h> #include <cudnn.h>
#include <nvrtc.h> #include <nvrtc.h>
//why use "do...while.." //why use "do...while.."
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment