Skip to content

Commit bb1c974

Browse files
author
hhsecond
committed
review clean up
1 parent f11dd94 commit bb1c974

File tree

4 files changed

+7
-9
lines changed

4 files changed

+7
-9
lines changed

CMakeLists.txt

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ ADD_DEFINITIONS(-DREDISMODULE_EXPERIMENTAL_API)
9797
IF(BUILD_TF)
9898
FIND_LIBRARY(TF_LIBRARIES NAMES tensorflow
9999
PATHS ${depsAbs}/libtensorflow/lib)
100-
MESSAGE(STATUS "Found TensorFlow Libraries: \"${TF_LIBRARIES}\")")
100+
MESSAGE(STATUS "Found TensorFlow Libraries: \"${TF_LIBRARIES}\")")
101101
IF (NOT TF_LIBRARIES)
102102
MESSAGE(WARNING "Could not find TensorFlow in ${depsAbs}/libtensorflow/lib. Trying find_package method")
103103
FIND_PACKAGE(TensorFlow REQUIRED)
@@ -113,11 +113,11 @@ ENDIF()
113113
#----------------------------------------------------------------------------------------------
114114

115115
IF(BUILD_TFLITE)
116-
FIND_LIBRARY(TFLITE_LIBRARIES_1 NAMES tensorflow-lite
117-
PATHS ${depsAbs}/libtensorflow-lite/lib)
118-
FIND_LIBRARY(TFLITE_LIBRARIES_2 NAMES benchmark-lib.a
119-
PATHS ${depsAbs}/libtensorflow-lite/lib)
120-
SET(TFLITE_LIBRARIES ${TFLITE_LIBRARIES_1} ${TFLITE_LIBRARIES_2})
116+
FIND_LIBRARY(TFLITE_LIBRARIES_1 NAMES tensorflow-lite
117+
PATHS ${depsAbs}/libtensorflow-lite/lib)
118+
FIND_LIBRARY(TFLITE_LIBRARIES_2 NAMES benchmark-lib.a
119+
PATHS ${depsAbs}/libtensorflow-lite/lib)
120+
SET(TFLITE_LIBRARIES ${TFLITE_LIBRARIES_1} ${TFLITE_LIBRARIES_2})
121121
MESSAGE(STATUS "Found TensorFlow Lite Libraries: \"${TFLITE_LIBRARIES}\")")
122122
IF (NOT TFLITE_LIBRARIES)
123123
MESSAGE(FATAL_ERROR "Could not find TensorFlow Lite")

get_deps.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,6 @@ if [[ $WITH_TFLITE != 0 ]]; then
170170
LIBTFLITE_ARCHIVE=libtensorflowlite-${TFLITE_OS}-${TFLITE_ARCH}-${TFLITE_VERSION}.tar.gz
171171

172172
[[ ! -f $LIBTFLITE_ARCHIVE || $FORCE == 1 ]] && wget $LIBTF_URL_BASE/$LIBTFLITE_ARCHIVE
173-
echo $LIBTF_URL_BASE/$LIBTFLITE_ARCHIVE
174173

175174
rm -rf $LIBTFLITE.x
176175
mkdir $LIBTFLITE.x

opt/build/libtorch/repack.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ elif [[ "$1" == "gpu" ]]; then
1616
else
1717
GPU=${GPU:-0}
1818
fi
19+
1920
OS=$(python3 $ROOT/opt/readies/bin/platform --os)
2021
ARCH=$(python3 $ROOT/opt/readies/bin/platform --arch)
2122

@@ -65,7 +66,6 @@ fi
6566
LIBTORCH_ZIP=libtorch-${PT_BUILD}-${PT_OS}-${PT_ARCH}-${PT_VERSION}.zip
6667
if [ ! -f $LIBTORCH_ARCHIVE ]; then
6768
echo "Downloading libtorch ${PT_VERSION} ${PT_BUILD}"
68-
echo $LIBTORCH_URL
6969
wget -q -O $LIBTORCH_ZIP $LIBTORCH_URL
7070
fi
7171

test/test_requirements.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
11
numpy
2-
scipy
32
scikit-image
43
redis-py-cluster

0 commit comments

Comments
 (0)