From a8af6c46d4a1db5b14d295a000d8fe57f104e074 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu>
Date: Thu, 3 Dec 2020 14:55:09 -0600
Subject: [PATCH] Fixing pool_mean translation bug
---
llvm/projects/keras/frontend/hpvm_dfg_translator.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/llvm/projects/keras/frontend/hpvm_dfg_translator.py b/llvm/projects/keras/frontend/hpvm_dfg_translator.py
index 233652600a..eda8ed9da8 100644
--- a/llvm/projects/keras/frontend/hpvm_dfg_translator.py
+++ b/llvm/projects/keras/frontend/hpvm_dfg_translator.py
@@ -166,7 +166,7 @@ class HPVMTranslator:
node_header_str += ", "
node_header_str += ") { \n"
- node_header_str += " __visc__hint(visc::CUDNN_TARGET); \n"
+ node_header_str += " __visc__hint(visc::PROMISE_TARGET); \n"
node_header_str += " __visc__attributes(" + str(num_params) + ", "
for i in range(num_params):
@@ -462,7 +462,7 @@ class HPVMTranslator:
if layer_type == "MaxPooling2D":
func_name = "__visc__tensor_pool_max"
if layer_type == "AveragePooling2D":
- func_name = "__visc__tensor_pool_avg"
+ func_name = "__visc__tensor_pool_mean"
inst_str += " void* r = " + func_name + "(t1, "
inst_str += str(pool_size[0]) + ", " + str(pool_size[1]) + ", "
--
GitLab