pytorch

1
""" build mode definitions for caffe2/caffe2 """
2

3
load("@fbcode//:BUILD_MODE.bzl", get_parent_modes = "all_modes_keep_gpu_sections_all_modes_use_lld")
4
load("@fbcode_macros//build_defs:create_build_mode.bzl", "extend_build_mode")
5

6
def update_mode_struct(name, mode_struct):
7
    if name == "dev":
8
        return extend_build_mode(
9
            mode_struct,
10
            # TODO(ipbrady): Modules introduce floating point inaccuracies (T43879333)
11
            cxx_modules = False,
12
        )
13
    else:
14
        return mode_struct
15

16
_modes = {
17
    mode_name: update_mode_struct(mode_name, mode_struct)
18
    for mode_name, mode_struct in get_parent_modes().items()
19
}
20

21
def get_modes():
22
    """ Return modes for this file """
23
    return _modes
24