[HIPIFY][Caffe2] Initial Caffe2 support

2019-02-23 20:46:22 +03:00
parent 31c4cc78b1
commit 2619f22e5c
13 changed files with 222 additions and 3 deletions
@@ -73,7 +73,8 @@ const char *apiNames[NUM_API_TYPES] = {
  "cuRAND API",
  "cuDNN API",
  "cuFFT API",
-  "cuSPARSE API"
+  "cuSPARSE API",
+  "CAFFE2 API"
 };

 namespace {