-
Notifications
You must be signed in to change notification settings - Fork 109
/
.bazelrc
70 lines (57 loc) · 1.65 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
65
66
67
68
69
70
# Bazel configuration for Yggdrasil Decision Forests
#
# OPTIONS
#
# Linux
#
# linux: Linux build.
#
# Windows
#
# windows: Windows build (unsupported).
#
# MacOS
#
# macos: MacOS build.
#
# Common flags.
common --experimental_repo_remote_exec
build --announce_rc
build -c opt
# Allow to use tensorflow internal methods.
build --nocheck_visibility
# Yggdrasil uses tensorflow for IO.
build --define=use_tensorflow_io=1
# Tensorflow Nightly uses absl::anyInvocable which is unsupported by the
# absl used by TF 2.12
build --define=use_absl_anyinvocable=1
# If 1, use the pre-compiled wrappers.
build --define=use_precompiled_wrappers=0
# The user can stop the training with Ctrl+C.
build --define=stop_training_on_interrupt=1
# Yggdrasil uses the tensorflow headers specialized for dynamic linking.
build --define=tensorflow_with_header_lib=1
# Nice print
build:linux --copt=-fdiagnostics-color=always
build --color=yes
# Suppress C++ compiler warnings.
build:linux --copt=-w
build:linux --host_copt=-w
build:macos --copt=-w
build:windows --copt=/W0
# Common build options.
build --define=framework_shared_object=true
build --enable_platform_specific_config
build --define=dynamic_loaded_kernels=true
build --copt=-DAUTOLOAD_DYNAMIC_KERNELS
build --spawn_strategy=standalone
build --strategy=Genrule=standalone
build --action_env TF_NEED_CUDA="0"
# Host specifics options.
build:macos --features=-supports_dynamic_linker
# On Intel machines, AVX2 is available
build:macos_intel --config=avx_linux
# Absl synchronization primitives currently do not work on MacOS.
# TODO
build:macos --define std_synchronization_primitives=1
build:windows --config=release_cpu_windows