UNPKG

react-native-caffe2

Version:

### Bring deep learning to mobile in 2 minutes 📱🥂

github.com/ltacker/react-native-caffe2

ltacker/react-native-caffe2

45 lines (33 loc) • 2.15 kB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
if((NOT TENSOR_INCLUDE_DIR) AND Eigen_SOURCE_DIR) # unless TENSOR_INCLUDE_DIR is defined, let's use current Eigen version set(TENSOR_INCLUDE_DIR ${Eigen_SOURCE_DIR}) set(TENSOR_FOUND TRUE) else() find_package(Tensor) endif() if (TENSOR_FOUND) include_directories(${TENSOR_INCLUDE_DIR}) btl_add_bench(btl_tensor_linear main_linear.cpp) btl_add_bench(btl_tensor_vecmat main_vecmat.cpp) btl_add_bench(btl_tensor_matmat main_matmat.cpp) btl_add_target_property(btl_tensor_linear COMPILE_FLAGS "-fno-exceptions -DBTL_PREFIX=tensor") btl_add_target_property(btl_tensor_vecmat COMPILE_FLAGS "-fno-exceptions -DBTL_PREFIX=tensor") btl_add_target_property(btl_tensor_matmat COMPILE_FLAGS "-fno-exceptions -DBTL_PREFIX=tensor") option(BTL_BENCH_NOGCCVEC "also bench Eigen explicit vec without GCC's auto vec" OFF) if(CMAKE_COMPILER_IS_GNUCXX AND BTL_BENCH_NOGCCVEC) btl_add_bench(btl_tensor_nogccvec_linear main_linear.cpp) btl_add_bench(btl_tensor_nogccvec_vecmat main_vecmat.cpp) btl_add_bench(btl_tensor_nogccvec_matmat main_matmat.cpp) btl_add_target_property(btl_tensor_nogccvec_linear COMPILE_FLAGS "-fno-exceptions -fno-tree-vectorize -DBTL_PREFIX=tensor_nogccvec") btl_add_target_property(btl_tensor_nogccvec_vecmat COMPILE_FLAGS "-fno-exceptions -fno-tree-vectorize -DBTL_PREFIX=tensor_nogccvec") btl_add_target_property(btl_tensor_nogccvec_matmat COMPILE_FLAGS "-fno-exceptions -fno-tree-vectorize -DBTL_PREFIX=tensor_nogccvec") endif() if(NOT BTL_NOVEC) btl_add_bench(btl_tensor_novec_linear main_linear.cpp OFF) btl_add_bench(btl_tensor_novec_vecmat main_vecmat.cpp OFF) btl_add_bench(btl_tensor_novec_matmat main_matmat.cpp OFF) btl_add_target_property(btl_tensor_novec_linear COMPILE_FLAGS "-fno-exceptions -DEIGEN_DONT_VECTORIZE -DBTL_PREFIX=tensor_novec") btl_add_target_property(btl_tensor_novec_vecmat COMPILE_FLAGS "-fno-exceptions -DEIGEN_DONT_VECTORIZE -DBTL_PREFIX=tensor_novec") btl_add_target_property(btl_tensor_novec_matmat COMPILE_FLAGS "-fno-exceptions -DEIGEN_DONT_VECTORIZE -DBTL_PREFIX=tensor_novec") endif(NOT BTL_NOVEC) endif (TENSOR_FOUND)