   }
 }
 
-#endif  // CONV2D_LAYER_CUH_
+#endif  // DATA_LAYER_CUH_


#ifndef CONV2D_LAYER_CUH_
#define CONV2D_LAYER_CUH_

#include <assert.h>
#include <stdio.h>

#include "utils/bitmap_image.hpp"
#include "basics/session.hpp"

// TODO: implement prefetch data in CPU memory
// TODO: implement prefetch data in another thread of CPU
// TODO: implement CUDA kernel for backward()

#define BLOCKDIM 32
