diff --git a/operator_contrib/MatmulLeakyReluCustom/AclNNInvocation/run.sh b/operator_contrib/MatmulLeakyReluCustom/AclNNInvocation/run.sh index 351ecfc374cc17cd02958e113371c91ae8608714..97520806d05e6fb646273c733293ede079d3d78b 100755 --- a/operator_contrib/MatmulLeakyReluCustom/AclNNInvocation/run.sh +++ b/operator_contrib/MatmulLeakyReluCustom/AclNNInvocation/run.sh @@ -82,7 +82,7 @@ function main { # 4. 运行可执行文件 cd $CURRENT_DIR/output echo "INFO: execute op!" - msprof op ./execute_matmul_leakyrelu_op + msprof --application="./execute_matmul_leakyrelu_op" if [ $? -ne 0 ]; then echo "ERROR: acl executable run failed! please check your project!" return 1 diff --git a/operator_contrib/MatmulLeakyReluCustom/AclNNInvocation/scripts/gen_data.py b/operator_contrib/MatmulLeakyReluCustom/AclNNInvocation/scripts/gen_data.py index 05099df29ae61147769e6da1ab6aff1d2eed0d62..10d72e241074faba47d58947186a499d6fb64b27 100644 --- a/operator_contrib/MatmulLeakyReluCustom/AclNNInvocation/scripts/gen_data.py +++ b/operator_contrib/MatmulLeakyReluCustom/AclNNInvocation/scripts/gen_data.py @@ -22,7 +22,7 @@ def gen_golden_data(): input_bias = np.random.randint(1, 10, [N]).astype(np.float32) alpha = 0.001 golden = (np.matmul(input_a.astype(np.float32), input_b.astype(np.float32))).astype(np.float32) - # golden = np.where(golden >= 0, golden, golden * alpha) + golden = np.where(golden >= 0, golden, golden * alpha) os.system("mkdir -p input") os.system("mkdir -p output") input_a.tofile("./input/input_a.bin")