diff --git a/hpvm/projects/torch2hpvm/torch2hpvm/template_hpvm.cpp.in b/hpvm/projects/torch2hpvm/torch2hpvm/template_hpvm.cpp.in index 0c1db9b1ff9d71cb9a8c8bbf3a2c64cec8331476..1c4a386ce2fd9e50953a49377df20c9d3ebf75da 100644 --- a/hpvm/projects/torch2hpvm/torch2hpvm/template_hpvm.cpp.in +++ b/hpvm/projects/torch2hpvm/torch2hpvm/template_hpvm.cpp.in @@ -90,6 +90,7 @@ int main(int argc, char *argv[]){ __hpvm__init(); startMemTracking(); + #pragma clang loop unroll(disable) for (int i = 0; i < batch_count; i++){ int start = i * batch_size, end = start + batch_size; copyInputBatch(input_path.c_str(), start, end, {{input_shape|join(', ')}}, {{input_name}});