@@ -33,9 +33,9 @@ void read_data(const char *filename, vector<unsigned long> &shape, vector<int32_
33
33
fscanf (fp, " %d " , &shape_dim);
34
34
printf (" shape_dim = %d\n " , shape_dim);
35
35
shape.resize (shape_dim);
36
- uint64_t size = 1 ;
36
+ uint32_t size = 1 ;
37
37
for (int i = 0 ; i < shape_dim; i++){
38
- int64_t value = 0 ;
38
+ int32_t value = 0 ;
39
39
fscanf (fp, " %ld " , &value);
40
40
shape[i] = value;
41
41
size *= shape[i];
@@ -88,7 +88,7 @@ struct OpArgs {
88
88
std::vector<DLTensor> args;
89
89
std::vector<CVMValue> arg_values;
90
90
std::vector<int > arg_tcodes;
91
- std::vector<int64_t > shape_data;
91
+ std::vector<int32_t > shape_data;
92
92
};
93
93
94
94
int run_LIF (string model_root, int device_type = 0 ) {
@@ -181,11 +181,11 @@ int run_LIF(string model_root, int device_type = 0) {
181
181
CHECK_STATUS (status, " free model failed" );
182
182
183
183
if (json_path.find (" yolo" ) != string::npos || json_path.find (" ssd" ) != string::npos) {
184
- uint64_t n_bytes = 4 ;
185
- uint64_t ns = output.size () / n_bytes;
184
+ uint32_t n_bytes = 4 ;
185
+ uint32_t ns = output.size () / n_bytes;
186
186
std::cout << " yolo output size = " << ns << " n_bytes = " << n_bytes << " \n " ;
187
187
int32_t * int32_output = static_cast <int32_t *>((void *)output.data ());
188
- for (auto i = 0 ; i < std::min<uint64_t >(60UL , ns); i++) {
188
+ for (auto i = 0 ; i < std::min<uint32_t >(60UL , ns); i++) {
189
189
std::cout << (int32_t )int32_output[i] << " " ;
190
190
if ((i + 1 ) % 6 == 0 )
191
191
std::cout << " \n " ;
@@ -201,7 +201,7 @@ int run_LIF(string model_root, int device_type = 0) {
201
201
std::cout << " \n " ;
202
202
} else {
203
203
std::cout << " output size = " << output.size () << " \n " ;
204
- for (auto i = 0 ; i < std::min<uint64_t >(6UL * 10 , output.size ()); i++) {
204
+ for (auto i = 0 ; i < std::min<uint32_t >(6UL * 10 , output.size ()); i++) {
205
205
std::cout << (int32_t )output[i] << " " ;
206
206
}
207
207
std::cout << " \n " ;
0 commit comments