Commit 443d023b by Philip Hyunsu Cho Committed by Yao Wang

Add a missing header in cuda_device_api.cc (#3621)

parent dedcf82f
...@@ -28,6 +28,7 @@ ...@@ -28,6 +28,7 @@
#include <tvm/runtime/registry.h> #include <tvm/runtime/registry.h>
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cstring>
#include "cuda_common.h" #include "cuda_common.h"
namespace tvm { namespace tvm {
......
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