forked from tensorflow/graphics
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add TensorFlow pip package's manylinux2010 toolchain to graphics so t…
…hat we can build custom TensorFlow ops for OpenGL rasterizer. PiperOrigin-RevId: 312554787
- Loading branch information
1 parent
832c607
commit 0593f0d
Showing
4 changed files
with
1,969 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,6 @@ | ||
# Preconfigured toolchain | ||
|
||
Preconfigured toolchain for building custom C++ ops for OpenGL rasterizer. | ||
|
||
Please use these toolchain configs with the Dockerfiles mentioned in | ||
[custom op repo](https://github.com/tensorflow/custom-op). |
175 changes: 175 additions & 0 deletions
175
third_party/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.1/BUILD.oss
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,175 @@ | ||
# This file is expanded from a template by cuda_configure.bzl | ||
# Update cuda_configure.bzl#verify_build_defines when adding new variables. | ||
|
||
load(":cc_toolchain_config.bzl", "cc_toolchain_config") | ||
|
||
licenses(["restricted"]) | ||
|
||
package(default_visibility = ["//visibility:public"]) | ||
|
||
toolchain( | ||
name = "toolchain-linux-x86_64", | ||
exec_compatible_with = [ | ||
"@bazel_tools//platforms:linux", | ||
"@bazel_tools//platforms:x86_64", | ||
], | ||
target_compatible_with = [ | ||
"@bazel_tools//platforms:linux", | ||
"@bazel_tools//platforms:x86_64", | ||
], | ||
toolchain = ":cc-compiler-local", | ||
toolchain_type = "@bazel_tools//tools/cpp:toolchain_type", | ||
) | ||
|
||
cc_toolchain_suite( | ||
name = "toolchain", | ||
toolchains = { | ||
"local|compiler": ":cc-compiler-local", | ||
"darwin|compiler": ":cc-compiler-darwin", | ||
"x64_windows|msvc-cl": ":cc-compiler-windows", | ||
"x64_windows": ":cc-compiler-windows", | ||
"arm": ":cc-compiler-local", | ||
"aarch64": ":cc-compiler-local", | ||
"k8": ":cc-compiler-local", | ||
"piii": ":cc-compiler-local", | ||
"ppc": ":cc-compiler-local", | ||
"darwin": ":cc-compiler-darwin", | ||
}, | ||
) | ||
|
||
cc_toolchain( | ||
name = "cc-compiler-local", | ||
all_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
ar_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
as_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
compiler_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
dwp_files = ":empty", | ||
linker_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
objcopy_files = ":empty", | ||
strip_files = ":empty", | ||
# To support linker flags that need to go to the start of command line | ||
# we need the toolchain to support parameter files. Parameter files are | ||
# last on the command line and contain all shared libraries to link, so all | ||
# regular options will be left of them. | ||
supports_param_files = 1, | ||
toolchain_config = ":cc-compiler-local-config", | ||
toolchain_identifier = "local_linux", | ||
) | ||
|
||
cc_toolchain_config( | ||
name = "cc-compiler-local-config", | ||
builtin_include_directories = [ | ||
"/dt7/usr/include/c++/7", | ||
"/dt7/usr/include/c++/7/x86_64-pc-linux-gnu", | ||
"/dt7/usr/include/c++/7/backward", | ||
"/dt7/usr/lib/gcc/x86_64-pc-linux-gnu/7/include", | ||
"/dt7/usr/lib/gcc/x86_64-pc-linux-gnu/7/include-fixed", | ||
"/dt7/usr/include", | ||
"/usr/local/cuda-10.1/targets/x86_64-linux/include", | ||
"/usr/local/cuda-10.1/include", | ||
"/usr/local/cuda-10.1/extras/CUPTI/include", | ||
"/usr/include", | ||
], | ||
builtin_sysroot = "", | ||
cpu = "local", | ||
cuda_path = "", | ||
extra_no_canonical_prefixes_flags = ["-fno-canonical-system-headers"], | ||
host_compiler_path = "clang/bin/crosstool_wrapper_driver_is_not_gcc", | ||
host_compiler_prefix = "/usr/bin", | ||
host_compiler_warnings = [], | ||
host_unfiltered_compile_flags = [], | ||
linker_bin_path = "/usr/bin", | ||
) | ||
|
||
cc_toolchain( | ||
name = "cc-compiler-darwin", | ||
all_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
ar_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
as_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
compiler_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
dwp_files = ":empty", | ||
linker_files = ":crosstool_wrapper_driver_is_not_gcc", | ||
objcopy_files = ":empty", | ||
strip_files = ":empty", | ||
supports_param_files = 0, | ||
toolchain_config = ":cc-compiler-local-darwin", | ||
toolchain_identifier = "local_darwin", | ||
) | ||
|
||
cc_toolchain_config( | ||
name = "cc-compiler-local-darwin", | ||
builtin_include_directories = [ | ||
"/dt7/usr/include/c++/7", | ||
"/dt7/usr/include/c++/7/x86_64-pc-linux-gnu", | ||
"/dt7/usr/include/c++/7/backward", | ||
"/dt7/usr/lib/gcc/x86_64-pc-linux-gnu/7/include", | ||
"/dt7/usr/lib/gcc/x86_64-pc-linux-gnu/7/include-fixed", | ||
"/dt7/usr/include", | ||
"/usr/local/cuda-10.1/targets/x86_64-linux/include", | ||
"/usr/local/cuda-10.1/include", | ||
"/usr/local/cuda-10.1/extras/CUPTI/include", | ||
"/usr/include", | ||
], | ||
cpu = "darwin", | ||
extra_no_canonical_prefixes_flags = ["-fno-canonical-system-headers"], | ||
host_compiler_path = "clang/bin/crosstool_wrapper_driver_is_not_gcc", | ||
host_compiler_prefix = "/usr/bin", | ||
host_compiler_warnings = [], | ||
host_unfiltered_compile_flags = [], | ||
linker_bin_path = "/usr/bin", | ||
) | ||
|
||
cc_toolchain( | ||
name = "cc-compiler-windows", | ||
all_files = ":windows_msvc_wrapper_files", | ||
ar_files = ":windows_msvc_wrapper_files", | ||
as_files = ":windows_msvc_wrapper_files", | ||
compiler_files = ":windows_msvc_wrapper_files", | ||
dwp_files = ":empty", | ||
linker_files = ":windows_msvc_wrapper_files", | ||
objcopy_files = ":empty", | ||
strip_files = ":empty", | ||
supports_param_files = 1, | ||
toolchain_config = ":cc-compiler-windows-config", | ||
toolchain_identifier = "local_windows", | ||
) | ||
|
||
cc_toolchain_config( | ||
name = "cc-compiler-windows-config", | ||
builtin_include_directories = [ | ||
"/dt7/usr/include/c++/7", | ||
"/dt7/usr/include/c++/7/x86_64-pc-linux-gnu", | ||
"/dt7/usr/include/c++/7/backward", | ||
"/dt7/usr/lib/gcc/x86_64-pc-linux-gnu/7/include", | ||
"/dt7/usr/lib/gcc/x86_64-pc-linux-gnu/7/include-fixed", | ||
"/dt7/usr/include", | ||
"/usr/local/cuda-10.1/targets/x86_64-linux/include", | ||
"/usr/local/cuda-10.1/include", | ||
"/usr/local/cuda-10.1/extras/CUPTI/include", | ||
"/usr/include", | ||
], | ||
cpu = "x64_windows", | ||
msvc_cl_path = "msvc_not_used", | ||
msvc_env_include = "msvc_not_used", | ||
msvc_env_lib = "msvc_not_used", | ||
msvc_env_path = "msvc_not_used", | ||
msvc_env_tmp = "msvc_not_used", | ||
msvc_lib_path = "msvc_not_used", | ||
msvc_link_path = "msvc_not_used", | ||
msvc_ml_path = "msvc_not_used", | ||
) | ||
|
||
filegroup( | ||
name = "empty", | ||
srcs = [], | ||
) | ||
|
||
filegroup( | ||
name = "crosstool_wrapper_driver_is_not_gcc", | ||
srcs = ["clang/bin/crosstool_wrapper_driver_is_not_gcc"], | ||
) | ||
|
||
filegroup( | ||
name = "windows_msvc_wrapper_files", | ||
srcs = glob(["windows/msvc_*"]), | ||
) |
Oops, something went wrong.