diff --git a/mace/kernels/opencl/cl/common.h b/mace/kernels/opencl/cl/common.h index 604100d5312f361df7fad2afedea2b3a321b818a..09731d77fe92f948e2b287ad13eb4e5f0b304ec3 100644 --- a/mace/kernels/opencl/cl/common.h +++ b/mace/kernels/opencl/cl/common.h @@ -32,8 +32,8 @@ #define CHECK_OUT_OF_RANGE_FOR_IMAGE2D(image, coord) #endif -#define READ_IMAGET(image, coord, value) \ - CMD_TYPE(read_image, CMD_DATA_TYPE)(image, coord, value) +#define READ_IMAGET(image, sampler, coord) \ + CMD_TYPE(read_image, CMD_DATA_TYPE)(image, sampler, coord) #define WRITE_IMAGET(image, coord, value) \ CHECK_OUT_OF_RANGE_FOR_IMAGE2D(image, coord) \ CMD_TYPE(write_image, CMD_DATA_TYPE)(image, coord, value); diff --git a/mace/python/tools/converter.py b/mace/python/tools/converter.py index 0f367da0fe81a9421cfdb200aea15cd9048310a4..8b43ce7fcf50c3034ac00c72c77bf3fca037eb55 100644 --- a/mace/python/tools/converter.py +++ b/mace/python/tools/converter.py @@ -175,7 +175,7 @@ def parse_args(): parser.add_argument( "--platform", type=str, default="tensorflow", help="tensorflow/caffe") parser.add_argument( - "--embed_model_data", type=str2bool, default=True, help="input shape.") + "--embed_model_data", type=str2bool, default=True, help="embed model data.") return parser.parse_known_args()