From 575af2482f6502d64f672575e1120eead01a26ef Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Thu, 25 Mar 2021 22:57:12 -0500
Subject: [PATCH] Adding pragma nounroll hint to batch loop (Keras frontend)
---
hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py b/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py
index fbe2f41c84..41d8b4a849 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"
--
GitLab