Commit a67753e7 authored by fenglei.tian's avatar fenglei.tian

remove scripts folder, change cude headers to <>

parent 3bbeaa31
...@@ -60,6 +60,7 @@ nervana_aeon.egg-info/ ...@@ -60,6 +60,7 @@ nervana_aeon.egg-info/
*.swo *.swo
build/ build/
scripts/
# makeenv and test intermediate files # makeenv and test intermediate files
tmp/ tmp/
......
...@@ -19,10 +19,9 @@ ...@@ -19,10 +19,9 @@
#include <functional> #include <functional>
#include <memory> #include <memory>
#include <vector> #include <vector>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cudnn.h> #include <cudnn.h>
#include "cublas_v2.h" #include <cublas_v2.h>
#include "ngraph/function.hpp" #include "ngraph/function.hpp"
#include "ngraph/runtime/call_frame.hpp" #include "ngraph/runtime/call_frame.hpp"
......
...@@ -22,11 +22,10 @@ ...@@ -22,11 +22,10 @@
#include <typeindex> #include <typeindex>
#include <typeinfo> #include <typeinfo>
#include <unordered_map> #include <unordered_map>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cudnn_v7.h> #include <cudnn_v7.h>
#include "cublas_v2.h" #include <cublas_v2.h>
#include "cuda.h" #include <cuda.h>
#include "ngraph/codegen/code_writer.hpp" #include "ngraph/codegen/code_writer.hpp"
#include "ngraph/codegen/compiler.hpp" #include "ngraph/codegen/compiler.hpp"
...@@ -232,8 +231,8 @@ void runtime::gpu::GPU_ExternalFunction::compile() ...@@ -232,8 +231,8 @@ void runtime::gpu::GPU_ExternalFunction::compile()
R"(// Generated by the NGraph GPU backend R"(// Generated by the NGraph GPU backend
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cudnn_v7.h> #include <cudnn_v7.h>
#include "cublas_v2.h" #include <cublas_v2.h>
#include "cuda.h" #include <cuda.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"
......
...@@ -22,8 +22,8 @@ ...@@ -22,8 +22,8 @@
#include <stdio.h> #include <stdio.h>
#include <string> #include <string>
#include "cuda.h" #include <cuda.h>
#include "cuda_runtime.h" #include <cuda_runtime.h>
#include "ngraph/runtime/gpu/gpu_util.hpp" #include "ngraph/runtime/gpu/gpu_util.hpp"
......
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