forked from Adlik/Adlik
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path.bazelrc
64 lines (40 loc) · 1.62 KB
/
.bazelrc
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
common --experimental_repo_remote_exec
build --cxxopt=-std=c++14
build --host_cxxopt=-std=c++14
build --define=grpc_no_ares=true
# OpenVINO.
build:openvino --define=runtime=openvino
# TensorFlow (CPU).
build:tensorflow-cpu --define=runtime=tensorflow
# GPU.
build:using_cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
build:using_cuda --define=using_cuda=true
build:using_cuda --define=using_cuda_nvcc=true
build:using_cuda --action_env=TF_NEED_CUDA=1
build:using_cuda --action_env=TF_CUDA_COMPUTE_CAPABILITIES=7.0
# TensorFlow (GPU).
build:tensorflow-gpu --define=runtime=tensorflow
build:tensorflow-gpu --define=tensorflow_enable_mlir_generated_gpu_kernels=1
build:tensorflow-gpu --config=using_cuda
build:tensorflow-gpu --copt=-DNDEBUG # https://github.com/tensorflow/tensorflow/issues/22766
# TensorFlow Lite (CPU).
build:tensorflow-lite-cpu --define=runtime=tensorflow-lite
build:arm --crosstool_top=@local_config_embedded_arm//:toolchain
build:arm --host_crosstool_top=@bazel_tools//tools/cpp:toolchain
build:aarch64 --config=arm
build:aarch64 --cpu=aarch64
# TensorRT.
build:tensorrt --define=runtime=tensorrt
build:tensorrt --config=using_cuda
#Tensorflow Tensorrt (GPU).
build:tensorflow-tensorrt --define=runtime=tensorflow-tensorrt
build:tensorflow-tensorrt --define=tensorflow_enable_mlir_generated_gpu_kernels=1
build:tensorflow-tensorrt --config=using_cuda
build:tensorflow-tensorrt --action_env=TF_NEED_TENSORRT=1
# ML runtime.
build:ml --define=runtime=ml
# TVM runtime.
build:tvm --define=runtime=tvm
# Paddle runtime
build:paddle --define=runtime=paddle
build:paddle --define=paddle_link_static=true