Merge pull request #111 from rohithkrn/caffe2-hipify-python

[Caffe2] Add db directory for hipifying

Author: ashishfarmer

caffe2/db/CMakeLists.txt

@@ -5,11 +5,13 @@ set(Caffe2_DB_COMMON_CPU_SRC
 set(Caffe2_DB_COMMON_GPU_SRC
     "${CMAKE_CURRENT_SOURCE_DIR}/create_db_op_gpu.cc"
 )
-
+set(Caffe2_DB_COMMON_HIP_SRC
+    "${CMAKE_CURRENT_SOURCE_DIR}/hip/create_db_op_hip.cc"
+)
 # Common files that are always going to be included.
 list(APPEND Caffe2_CPU_SRCS ${Caffe2_DB_COMMON_CPU_SRC})
 list(APPEND Caffe2_GPU_SRCS ${Caffe2_DB_COMMON_GPU_SRC})
-
+list(APPEND Caffe2_HIP_SRCS ${Caffe2_DB_COMMON_HIP_SRC})
 # DB specific files
 if (USE_LMDB)
   list(APPEND Caffe2_CPU_SRCS "${CMAKE_CURRENT_SOURCE_DIR}/lmdb.cc")
@@ -25,3 +27,5 @@ endif()
 
 set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE)
 set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} PARENT_SCOPE)
+set(Caffe2_HIP_SRCS ${Caffe2_HIP_SRCS} PARENT_SCOPE)
+

tools/amd_build/build_caffe2_amd.py

@@ -17,6 +17,7 @@
     "caffe2/queue/*",
     "caffe2/**/*_test*",
     "caffe2/core/THCCachingAllocator*",
+    "caffe2/db/*",
 ]
 
 ignores = [