fix a bug

hijkzzz · May 13, 2019 · 528ce8d · 528ce8d
1 parent 7e17879
commit 528ce8d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -1,6 +1,7 @@
 #pragma once
 
 #include <storage.cuh>
+#include <utils.cuh>
 
 #include <memory>
 #include <vector>