From 528ce8dd01c64fe49b1d19dc73b760600b277769 Mon Sep 17 00:00:00 2001
From: hijkzzz <janhu9527@gmail.com>
Date: Mon, 13 May 2019 09:36:59 +0800
Subject: [PATCH] fix a bug

---
 src/cuda/conv.cu   | 4 ++--
 src/cuda/layer.cuh | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/cuda/conv.cu b/src/cuda/conv.cu
index d37894b..94d44ae 100644
--- a/src/cuda/conv.cu
+++ b/src/cuda/conv.cu
@@ -1,4 +1,4 @@
-#include <conv.cuh>
+#include <conv.cuh>
 
 #include <thrust/copy.h>
 #include <thrust/device_vector.h>
@@ -238,7 +238,7 @@ void operator_d_conv(
   // dL/d_col = F^T * dL/dY
   std::vector<int> dl_dcol_shape{batch_size, channel_in * kernel_h * kernel_w,
                                  height_col * width_col};
-  INIT_TEMP(temp, "dl_dcol", dl_df_shape);
+  INIT_TEMP(temp, "dl_dcol", dl_dcol_shape);
   operator_matmul(temp["filters_t"].get(), outputs_grad, temp["dl_dcol"].get(),
                   1);  // broadcast param 1
 
diff --git a/src/cuda/layer.cuh b/src/cuda/layer.cuh
index 7b58cb1..7e989fb 100644
--- a/src/cuda/layer.cuh
+++ b/src/cuda/layer.cuh
@@ -1,6 +1,7 @@
 #pragma once
 
 #include <storage.cuh>
+#include <utils.cuh>
 
 #include <memory>
 #include <vector>