From a96ba017900793f95f268df80e8c9a669f79b865 Mon Sep 17 00:00:00 2001 From: denghuilu Date: Wed, 1 Dec 2021 23:00:31 +0800 Subject: [PATCH] fix bug of hip model compression --- source/lib/src/rocm/tabulate.hip.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/source/lib/src/rocm/tabulate.hip.cu b/source/lib/src/rocm/tabulate.hip.cu index 6b6270c18f..10bc849db4 100644 --- a/source/lib/src/rocm/tabulate.hip.cu +++ b/source/lib/src/rocm/tabulate.hip.cu @@ -426,7 +426,7 @@ __global__ void tabulate_fusion_se_t_grad_grad_fifth_order_polynomial( const FPTYPE max, const FPTYPE stride0, const FPTYPE stride1, - const int nnei, + const int nnei_i, const int nnei_j, const int last_layer_size) {