Commit Graph

80 Commits

Author SHA1 Message Date
Amit Patankar
8173e6a75b Updating to bazel 0.15.0. 2018-07-26 11:47:37 -07:00
Toby Boyd
9baff09279 internal change
PiperOrigin-RevId: 204832902
2018-07-16 18:02:25 -07:00
A. Unique TensorFlower
bdceb78765 Disable more TF_NEED_* on Windows by default.
This simplifies ./configure process on Windows, we can remove them when we actually support the corresponding feature on Windows.

PiperOrigin-RevId: 204804112
2018-07-16 18:02:05 -07:00
Smit Hinsu
5066b248e1 Update default cuda compute capability while installing from sources to cover up to Volta
PiperOrigin-RevId: 204606836
2018-07-16 18:02:00 -07:00
Smit Hinsu
63e6b9bf43 Update default NCCL version while installing from source to 2.2
This is to keep the default configuration consistent with prebuilt TensorFlow.

PiperOrigin-RevId: 204513386
2018-07-13 12:50:24 -07:00
A. Unique TensorFlower
ab39198ace Add CROSSTOOL for Windows GPU build with nvcc compiler
After this change, to build TensorFlow with GPU support on Windows, you just need to follow the same procedure as Linux.

Also re-enable remote cache since the bug in b/111106393 has been fixed by the new wrapper script.

The clean-up of the wrapper scripts also made the build around 5 mins faster than using Bazel's msvc wrapper script.

PiperOrigin-RevId: 204105368
2018-07-11 04:59:54 -07:00
A. Unique TensorFlower
02f17fe8b7 Set correct environment variable for CUDA install path
configure.py respects CUDA_TOOLKIT_PATH instead of CUDA_INSTALL_PATH

PiperOrigin-RevId: 203591214
2018-07-08 17:14:19 -07:00
Yong Tang
a7b7aa856f Add KinesisDataset support for tensorflow Dataset (#19712)
* Add KinesisDataset support for tensorflow Dataset

This fix is an attempt to add Kinesis support
for tensorflow's Dataset. Kinesis is provided by
AWS as a managed data streaming service. It is
similiar to Apache Kafka, often used in places
where maintaining a independent Kafka cluster on AWS
is not desirable or possible.

This fix adds the Kinesis support for tensorflow Dataset.
Similiar to the Kafka integration in tensorflow,
KinesisDataset outputs tf.string for records.

Test cases have also been added, which could be invoked manually.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Expose KinesisDataset in dataset_ops.cc

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Expose KinesisDataset in python wrapper

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Add test cases for KinesisDataset

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Update AWS library include files

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Add Bazel BUILD files

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Rename s3_crypto to aws_crypto

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Rename with_s3_support to with_aws_support

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Selectively add kinesis to tensorflow/contrib/BUILD

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Set different partition key and pylint fix.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Add missing modules in cmake's python_modules.txt

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

* Address review feedback

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
2018-07-02 07:41:42 -07:00
A. Unique TensorFlower
f70dfea26f Checking that the NCCL 2 license file is found, see GitHub issue 19679.
PiperOrigin-RevId: 202613754
2018-06-29 01:48:36 -07:00
Mingxing Tan
1e7b0e4ad6 Merge changes from github.
PiperOrigin-RevId: 202585094
2018-06-28 19:16:41 -07:00
Akshay Modi
6070ae0e14 Merge changes from github.
PiperOrigin-RevId: 201110240
2018-06-18 21:04:00 -07:00
Akshay Modi
148b4381fd Automated g4 rollback of changelist 201011811
PiperOrigin-RevId: 201033171
2018-06-18 11:55:03 -07:00
Akshay Modi
e80732c989 Merge changes from github.
PiperOrigin-RevId: 201011811
2018-06-18 09:59:59 -07:00
Michael Case
5105350be9 Moves generated android_sdk() and android_ndk() repo rules out of WORKSPACE.
These rules currently get written by configure.py script to WORKSPACE
file which is not ideal since (1) WORKSPACE file is tracked by git and
(2) we require users to manually delete the rules in order to
update/regenerate them.

Moving these rules into an external repo that is generated based on
several ENV variables set by the configure.py script. Modifying any
of these ENV variables will cause the rules to be updated.

PiperOrigin-RevId: 199388460
2018-06-05 17:50:28 -07:00
Yifei Feng
b59833c3fd Merge changes from github.
Revert #18413. Too many internal test failures due to the name scope change caused by this change.
Revert #18192. Cannot use re2::StringPiece internally. Need alternative for set call. Will pull and clean this up in a separate change.

PiperOrigin-RevId: 197991247
2018-05-24 19:15:01 -07:00
Michael Case
b39f1f5430 Remove -DGEMMLOWP_ALLOW_SLOW_SCALAR_FALLBACK being added by default to all builds.
PiperOrigin-RevId: 197037867
2018-05-17 13:04:48 -07:00
A. Unique TensorFlower
b15500be31 Remove outdated CUDA SDK string (the text is now consistent with other version choices, and the '9.0' format is already present in the default).
PiperOrigin-RevId: 195855416
2018-05-08 15:43:55 -07:00
Michael Case
7ee54c2f7c Remove deprecated/unused python related Bazel options.
Since py_runtime was introduced, Bazel ignores options such as
--force_python2 and --python2_path. Deleting to clean stuff up and
make sure people are not misled.

PiperOrigin-RevId: 193127681
2018-04-16 17:38:36 -07:00
Michael Case
0073d1375a Fix Windows GPU TensorFlow Bazel builds.
The configure.py script will error out on Windows GPU builds due
to NCCL attempted to be configured (and is currently Linux only).

PiperOrigin-RevId: 192461362
2018-04-11 09:41:42 -07:00
Michael Case
5ad9e45888 Merge changes from github.
PiperOrigin-RevId: 192388250
2018-04-10 18:46:38 -07:00
A. Unique TensorFlower
1fda7645d1 Add support for NCCL2. The configure script asks for what version of NCCL to use. The default is still NCCL 1 from GitHub. If the user chooses NCCL 2, it asks for the install directory.
The nccl_configure.bzl generates two different BUILD files based on the chose NCCL version. For NCCL 1, it aliases to the existing 'nccl_archive' http_repo on GitHub. For NCCL 2, it creates a target containing the NCCL 2 library and headers from the chosen install directory.

PiperOrigin-RevId: 191718007
2018-04-05 03:11:33 -07:00
A. Unique TensorFlower
1bba94af33 Compile TensorFlow with /arch:AVX on Windows
/arch:AVX is the correpsonding option in MSVC for gcc's --march=native

PiperOrigin-RevId: 191660389
2018-04-04 15:51:49 -07:00
Jianwei Xie
63dffd5a3b Automated g4 rollback of changelist 190858242
PiperOrigin-RevId: 190953197
2018-03-29 10:52:55 -07:00
Anna R
108178da2a Automated g4 rollback of changelist 190835392
PiperOrigin-RevId: 190858242
2018-03-28 16:55:15 -07:00
Jianwei Xie
e97c9e91e0 Merge changes from github.
PiperOrigin-RevId: 190835392
2018-03-28 14:38:46 -07:00
Benoit Steiner
0dadbfe118 Merge changes from github.
PiperOrigin-RevId: 190161440
2018-03-22 18:57:11 -07:00
Ilya Biryukov
9e651e4571 Allow to download clang and use clang for CPU builds.
Previously we only allowed to download clang when doing GPU builds.

The added skylark files use bazel's autoconf scripts, which were only added in
0.10.0. To provide nice error message for older versions of bazel
(i.e. 'version is less than 0.10' vs 'can't load @bazel_tools/cpp/...'), we
move the bazel version check into WORKSPACE file from workspace.bzl.

PiperOrigin-RevId: 190050798
2018-03-22 05:36:18 -07:00
Jacques Pienaar
2d0531d72c Merge changes from github.
PiperOrigin-RevId: 189945839
2018-03-21 12:10:30 -07:00
A. Unique TensorFlower
1e75c69339 Automated g4 rollback of changelist 189231636
PiperOrigin-RevId: 189258641
2018-03-15 15:49:01 -07:00
Jacques Pienaar
ccd8079e57 Merge changes from github.
PiperOrigin-RevId: 189231636
2018-03-15 13:03:19 -07:00
Shanqing Cai
7144571f2f Merge changes from github.
PiperOrigin-RevId: 188817194
2018-03-12 19:37:39 -07:00
Yifei Feng
dce9a49c19 Merge changes from github.
PiperOrigin-RevId: 186674197
2018-02-22 14:29:27 -08:00
Ankur Taly
0e6f39d1bd Merge changes from github.
PiperOrigin-RevId: 186073337
2018-02-16 18:27:19 -08:00
Michael Case
d90054e7c0 Merge changes from github.
PiperOrigin-RevId: 184897758
2018-02-07 14:39:49 -08:00
Guangda Lai
76f6938baf Set up TensorRT configurations for external use, and add a test.
PiperOrigin-RevId: 183347199
2018-01-26 00:02:44 -08:00
Frank Chen
c4ef927b5e Merge changes from github.
PiperOrigin-RevId: 181494416
2018-01-10 11:49:30 -08:00
Michael Case
d31531af15 Move some build configs to tools/bazel.rc from configure.py
Moving --config=android_arm --config=mkl and --config=monolithic
build configs into tools/bazel.rc. These options are just always
written the same way to .bazelrc when configure.py is run. This
should trim down the scope of configure.py and make it easier to
build TF without running configure.py prior.

PiperOrigin-RevId: 180973131
2018-01-05 14:13:39 -08:00
Michael Case
a64485dbb3 Run gen_git_source.py inside of a repo_rule instead of configure.
PiperOrigin-RevId: 179971055
2017-12-22 15:48:23 -08:00
A. Unique TensorFlower
e4532d2097 Merge changes from github.
PiperOrigin-RevId: 179953488
2017-12-22 12:46:28 -08:00
A. Unique TensorFlower
fd29e9534f Added an option to download a fresh release of clang for doing cuda_clang builds.
Tested only on Linux, but should also work for Mac.
No support for Windows is available yet.

PiperOrigin-RevId: 179910980
2017-12-22 03:11:39 -08:00
Dandelion Man?
90e42f3ac8 Automated g4 rollback of changelist 179260538
PiperOrigin-RevId: 179263865
2017-12-15 18:19:09 -08:00
A. Unique TensorFlower
9648f8040a Automated g4 rollback of changelist 179258973
PiperOrigin-RevId: 179260538
2017-12-15 17:39:26 -08:00
Dandelion Man?
d55f532867 Merge changes from github.
PiperOrigin-RevId: 179258973
2017-12-15 17:16:29 -08:00
Austin Anderson
b873363188 Fix duplicated trisycl_include_dir
A recent botched three-way-merge sync seems to have duplicated (more than once on Github) the set_trisycl_include_dir function. This change removes the duplicates in favor of a cleaned-up variant that I added in another recent change.

PiperOrigin-RevId: 179249451
2017-12-15 15:48:21 -08:00
Austin Anderson
f9a88f8a2b Cap prompt attempts before aborting, like other prompts
PiperOrigin-RevId: 178936195
2017-12-13 11:53:40 -08:00
Shanqing Cai
fe8406149f Merge changes from github.
PiperOrigin-RevId: 178185697
2017-12-06 18:47:41 -08:00
Austin Anderson
6affacedbb Add android rule helpers and cleanup input loops
This change teaches the configure script how to search for Android NDK
and SDK installations and create new WORKSPACE rules pointing to them.
It also refactors many similar loop-over-user-input functions into using
a reusable method (not the more complex ones).

Specifying an SDK directory will further query for the available SDK API
levels and build tools versions, but it won't perform any compatibility
checks.

Like other settings, every android-related setting can be set beforehand
via an env param. The script will not ask for any Android settings if
there are already any android repository rules in the WORKSPACE.

The script will emit a warning if using an NDK version newer than 14 due
to https://github.com/bazelbuild/bazel/issues/4068.

PiperOrigin-RevId: 177989785
2017-12-05 12:04:55 -08:00
Yifei Feng
b1d8c59e9b Merge changes from github.
PiperOrigin-RevId: 176695926
2017-11-22 13:50:02 -08:00
Yifei Feng
e70c00950d Automated g4 rollback of changelist 176615107
PiperOrigin-RevId: 176622438
2017-11-22 00:42:30 -08:00
Yifei Feng
ad7eeec1cc Automated g4 rollback of changelist 176615737
PiperOrigin-RevId: 176621645
2017-11-22 00:36:36 -08:00