diff --git a/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py b/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py index fbe2f41c84c2d018412a682b3c211ad9897dcd5f..41d8b4a8491ced3e8281164aaf7ea49d3ec68551 100644 --- a/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py +++ b/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py @@ -643,6 +643,7 @@ class HPVMTranslator: func_str += "startProfiling(); \n\n" func_str += "for(unsigned int j = 0; j < 1; j++){ \n" + func_str += "#pragma clang loop unroll(disable) \n" func_str += "for(unsigned int i = 0; i < batch_count; i++){ \n\n" func_str += "unsigned int start = i * batch_size; \n"