diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/alexnet_cifar10.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/alexnet_cifar10.cc
index 68bf89546e71cf5e07e27ef4141fbe8b42818ee5..8129fbfafcdd3e991e67d33fd3013e1700da45c5 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/alexnet_cifar10.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/alexnet_cifar10.cc
@@ -13,7 +13,7 @@ int main(){
llvm_hpvm_initTensorRt(0);
- std::string dir_prefix = std::string("../model_params/alexnet_cifar10_test/");
+ std::string dir_prefix = std::string("../model_params/alexnet_cifar10/");
std::string input_path = dir_prefix + std::string("input.bin");
std::string labels_path = dir_prefix + std::string("labels.bin");
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/half/alexnet_cifar10_half.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/half/alexnet_cifar10_half.cc
index 8555518570dd24c244e97f8d86452e74eaffb47c..8a429862f34f95793dd9ca7caa619b10dbe568ab 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/half/alexnet_cifar10_half.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/half/alexnet_cifar10_half.cc
@@ -14,7 +14,7 @@ int main(){
llvm_hpvm_initTensorRt(0);
- std::string dir_prefix = std::string("../model_params/alexnet_cifar10_test/");
+ std::string dir_prefix = std::string("../model_params/alexnet_cifar10/");
std::string input_path = dir_prefix + std::string("input.bin");
std::string labels_path = dir_prefix + std::string("labels.bin");
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet_promise.cc
index d0f4a3416788cae45b06199c16826ee9c1955aab..c67eb1153e6c29a0f478e495be2d36dbdafe1d56 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet_promise.cc
@@ -55,7 +55,7 @@ int main(int argc, char* argv[]){
}
- llvm_hpvm_initTensorRt(0);
+ llvm_hpvm_initTensorRt(1);
int missed = 0;
@@ -70,7 +70,7 @@ int main(int argc, char* argv[]){
int batch_count = test_input_size / batch_size;
float final_accuracy = 0.0;
- std::string dir_prefix = std::string("../model_params/alexnet_cifar10_test/");
+ std::string dir_prefix = std::string("../model_params/alexnet_cifar10/");
std::string input_path = dir_prefix + std::string("input.bin");
std::string labels_path = dir_prefix + std::string("labels.bin");
std::string labels32_path = dir_prefix + std::string("labels32.bin");
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/approxhpvm_src.cc b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/approxhpvm_src.cc
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/approxhpvm_src.cc
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/approxhpvm_src.cc
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_1_b.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_1_b.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_1_b.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_1_b.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_1_w.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_1_w.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_1_w.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_1_w.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_2_b.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_2_b.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_2_b.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_2_b.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_2_w.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_2_w.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_2_w.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_2_w.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_3_b.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_3_b.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_3_b.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_3_b.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_3_w.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_3_w.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_3_w.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_3_w.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_4_b.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_4_b.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_4_b.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_4_b.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_4_w.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_4_w.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_4_w.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_4_w.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_5_b.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_5_b.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_5_b.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_5_b.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_5_w.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_5_w.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/conv2d_5_w.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/conv2d_5_w.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/dense_1_b.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/dense_1_b.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/dense_1_b.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/dense_1_b.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/dense_1_w.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/dense_1_w.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/dense_1_w.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/dense_1_w.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/input.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/input.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/input.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/input.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/labels.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/labels.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/labels.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/labels.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/labels32.bin b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/labels32.bin
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/labels32.bin
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/labels32.bin
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/layer_composition.txt b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/layer_composition.txt
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/layer_composition.txt
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/layer_composition.txt
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/layers.txt b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/layers.txt
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/layers.txt
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/layers.txt
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/promise_src.cc b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/promise_src.cc
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/promise_src.cc
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/promise_src.cc
diff --git a/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/src.cc b/llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/src.cc
similarity index 100%
rename from llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/src.cc
rename to llvm/projects/hpvm-tensor-rt/model_params/alexnet_cifar10/src.cc
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet.cpp
index 7b44c7699cecb8d7bada8c68ebf66cfc16322a7e..897020b056272e3136e9c2e777ab3eb07df03bd2 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet.cpp
@@ -9,6 +9,8 @@
#include <tensorTypes.h>
#include <tensorUtils.h>
+
+
void var_0_node(void* t1, size_t bytes_t1, void* t2, size_t bytes_t2) {
__visc__hint(visc::CUDNN_TARGET);
__visc__attributes(2, t1, t2, 0);
@@ -177,6 +179,8 @@ void var_20_node(void* t1, size_t bytes_t1) {
__visc__return(2, r, (size_t) 0);
}
+
+
void root(void* input, size_t input_bytes,
void* conv2d_1_w, size_t conv2d_1_w_bytes,
void* conv2d_1_b, size_t conv2d_1_b_bytes,
@@ -192,7 +196,8 @@ void root(void* input, size_t input_bytes,
void* dense_1_b, size_t dense_1_b_bytes){
- __visc__hint(visc::CPU_TARGET);
+ __visc__hint(visc::CPU_TARGET);
+
__visc__attributes(13, input, conv2d_1_w, conv2d_1_b, conv2d_2_w, conv2d_2_b, conv2d_3_w, conv2d_3_b, conv2d_4_w, conv2d_4_b, conv2d_5_w, conv2d_5_b, dense_1_w, dense_1_b, 0);
@@ -369,15 +374,12 @@ RootIn;
int main(){
- std::string dir_prefix = std::string("../../../../../../projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/");
+ std::string dir_prefix = std::string("../../../../../../projects/hpvm-tensor-rt/model_params/alexnet_cifar10/");
std::string input_path = dir_prefix + std::string("input.bin");
- //void* input = readTrainedWeights(input_path.c_str(), 0,10000,3,32,32);
- void* input = create4DTensor(0,nchw,10000,3,32,32);
- copyInputBatch(input_path.c_str(),0,10000,3,32,32, input);
-
+ void* input = readTrainedWeights(input_path.c_str(), 0,5000,3,32,32);
std::string labels_path = dir_prefix + std::string("labels.bin");
- uint8_t* labels = readLabels(labels_path.c_str(),10000);
+ uint8_t* labels = readLabels(labels_path.c_str(),5000);
std::string conv2d_1_w_path = dir_prefix + std::string("conv2d_1_w.bin");
void* conv2d_1_w = readTrainedWeights(conv2d_1_w_path.c_str(), 0,64,3,11,11);
std::string conv2d_1_b_path = dir_prefix + std::string("conv2d_1_b.bin");
@@ -443,7 +445,7 @@ int main(){
hpvm_request_tensor(result, 0);
__visc__cleanup();
- computeAccuracy2(labels, 10000, result);
+ computeAccuracy2(labels, 5000, result);
return 0;
}
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp
index 82dfcb725b6e69f49a4fef88037c544dcf257ed1..09e6e25cffe18e375575bf83ebd4283e4295e7e4 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp
@@ -373,7 +373,7 @@ RootIn;
int main(){
- std::string dir_prefix = std::string("../../../../../../projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/");
+ std::string dir_prefix = std::string("../../../../../../projects/hpvm-tensor-rt/model_params/alexnet_cifar10/");
std::string input_path = dir_prefix + std::string("input.bin");
//void* input = readTrainedWeights(input_path.c_str(), 0,5000,3,32,32);
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_promise.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_promise.cpp
index 9bff465223385ca01720d6cc0a2e20e711eaf838..a7a2a9efba0c1e7c0e815b594db772159d9aa4be 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_promise.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_promise.cpp
@@ -373,7 +373,7 @@ RootIn;
int main(){
- std::string dir_prefix = std::string("../../../../../../projects/hpvm-tensor-rt/model_params/alexnet_cifar10_test/");
+ std::string dir_prefix = std::string("../../../../../../projects/hpvm-tensor-rt/model_params/alexnet_cifar10/");
std::string input_path = dir_prefix + std::string("input.bin");
void* input = readTrainedWeights(input_path.c_str(), 0,5000,3,32,32);