From 24d6f7db31dd093fa39f0a516ef4c3893a80172d Mon Sep 17 00:00:00 2001 From: chenzupeng Date: Fri, 4 Sep 2020 10:23:48 +0800 Subject: [PATCH] GPU CI: add OCR02 and OCR08 --- mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl | 8 ++++---- mindspore/lite/test/models_tflite_gpu.cfg | 4 +++- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl b/mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl index 4b07a1e89e0..791ba888eb1 100644 --- a/mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl +++ b/mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl @@ -58,10 +58,10 @@ __kernel void transpose_NHWC4_BUF(__read_only image2d_t src_data, global FLT4 *d bool over_size = W * C.y > 65535; FLT4 x0, x1, x2, x3; if (over_size) { - x0 = READ_IMAGE(src_data, smp_zero, (int2)(C, 4 * X)); - x1 = READ_IMAGE(src_data, smp_zero, (int2)(C, 4 * X + 1)); - x2 = READ_IMAGE(src_data, smp_zero, (int2)(C, 4 * X + 2)); - x3 = READ_IMAGE(src_data, smp_zero, (int2)(C, 4 * X + 3)); + x0 = READ_IMAGE(src_data, smp_zero, (int2)(Y, 4 * X)); + x1 = READ_IMAGE(src_data, smp_zero, (int2)(Y, 4 * X + 1)); + x2 = READ_IMAGE(src_data, smp_zero, (int2)(Y, 4 * X + 2)); + x3 = READ_IMAGE(src_data, smp_zero, (int2)(Y, 4 * X + 3)); } else { x0 = READ_IMAGE(src_data, smp_zero, (int2)((4 * X) % W * C.y + Y, (4 * X) / W)); x1 = READ_IMAGE(src_data, smp_zero, (int2)((4 * X + 1) % W * C.y + Y, (4 * X + 1) / W)); diff --git a/mindspore/lite/test/models_tflite_gpu.cfg b/mindspore/lite/test/models_tflite_gpu.cfg index d432b29fef9..ac008b9b182 100644 --- a/mindspore/lite/test/models_tflite_gpu.cfg +++ b/mindspore/lite/test/models_tflite_gpu.cfg @@ -1,3 +1,5 @@ mobilenet_v1_1.0_224.tflite mobilenet_v2_1.0_224.tflite -resnet.tflite \ No newline at end of file +resnet.tflite +hiai_cn_recognize_modify_padv2.tflite +hiai_cv_focusShootOCRModel_08.tflite