Browse Source

add int8 clip

tags/v1.6.0
gongdaguo 4 years ago
parent
commit
a2afb88f0d
2 changed files with 7 additions and 12 deletions
  1. +7
    -7
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/CMakeLists.txt
  2. +0
    -5
      mindspore/lite/CMakeLists.txt

+ 7
- 7
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/CMakeLists.txt View File

@@ -35,13 +35,7 @@ file(GLOB KERNEL_SRC
${NNACL_DIR}/fp32_grad/*.c
)

if(OP_INT8_CLIP)
set(KERNEL_SRC
${KERNEL_SRC}
${NNACL_DIR}/int8/pack_int8.c
${NNACL_DIR}/int8/quantize.c
)
else()
if((NOT DEFINED MSLITE_ENABLE_INT8) OR MSLITE_ENABLE_INT8)
file(GLOB KERNEL_SRC_INT8
${NNACL_DIR}/int8/*.c
)
@@ -49,6 +43,12 @@ else()
${KERNEL_SRC}
${KERNEL_SRC_INT8}
)
else()
set(KERNEL_SRC
${KERNEL_SRC}
${NNACL_DIR}/int8/pack_int8.c
${NNACL_DIR}/int8/quantize.c
)
endif()

if(MSLITE_ENABLE_SPARSE_COMPUTE)


+ 0
- 5
mindspore/lite/CMakeLists.txt View File

@@ -100,11 +100,6 @@ endif()
if(DEFINED ENV{MSLITE_ENABLE_INT8})
set(MSLITE_ENABLE_INT8 $ENV{MSLITE_ENABLE_INT8})
endif()
if(MSLITE_ENABLE_INT8)
set(OP_INT8_CLIP off)
else()
set(OP_INT8_CLIP on)
endif()
if(DEFINED ENV{MSLITE_ENABLE_SPARSE_COMPUTE})
set(MSLITE_ENABLE_SPARSE_COMPUTE $ENV{MSLITE_ENABLE_SPARSE_COMPUTE})
endif()


Loading…
Cancel
Save