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 @@
#include <cublas_v2.h>
#include <cuda.h>
#include <cuda_runtime.h>
#include <cudnn_v7.h>
#include <cudnn.h>
#include "ngraph/axis_set.hpp"
#include "ngraph/runtime/gpu/gpu_runtime_context.hpp"
......
......@@ -19,7 +19,7 @@
#include <cublas_v2.h>
#include <cuda.h>
#include <cuda_runtime.h>
#include <cudnn_v7.h>
#include <cudnn.h>
#include <iostream>
#include <nvrtc.h>
#include <string>
......
......@@ -18,7 +18,7 @@
#include <cublas_v2.h>
#include <cuda.h>
#include <cuda_runtime.h>
#include <cudnn_v7.h>
#include <cudnn.h>
#include <fstream>
#include <memory>
#include <string>
......@@ -287,7 +287,7 @@ void runtime::gpu::GPU_ExternalFunction::compile()
#include <cublas_v2.h>
#include <cuda.h>
#include <cuda_runtime.h>
#include <cudnn_v7.h>
#include <cudnn.h>
#include "ngraph/descriptor/input.hpp"
#include "ngraph/descriptor/layout/dense_tensor_view_layout.hpp"
......
......@@ -24,7 +24,7 @@
#include <cublas_v2.h>
#include <cuda.h>
#include <cuda_runtime.h>
#include <cudnn_v7.h>
#include <cudnn.h>
#include <nvrtc.h>
//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