3
0
Fork 0
forked from mirrors/nixpkgs

Merge pull request #76945 from prusnak/tensorflow-drop-bin2c-path

python.pkgs.tensorflow: fix build with cudatoolkit 10.2
This commit is contained in:
Timo Kaufmann 2020-01-06 17:12:34 +01:00 committed by GitHub
commit bdbec90304
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 0 deletions

View file

@ -0,0 +1,10 @@
--- a/third_party/nccl/build_defs.bzl.tpl
+++ b/third_party/nccl/build_defs.bzl.tpl
@@ -113,7 +113,6 @@ def _device_link_impl(ctx):
"--cmdline=--compile-only",
"--link",
"--compress-all",
- "--bin2c-path=%s" % bin2c.dirname,
"--create=%s" % tmp_fatbin.path,
"--embedded-fatbin=%s" % fatbin_h.path,
] + images,

View file

@ -124,6 +124,10 @@ let
sha256 = "077cpj0kzyqxzdya1dwh8df17zfzhqn7c685hx6iskvw2979zg2n";
})
./lift-gast-restriction.patch
# cuda 10.2 does not have "-bin2c-path" option anymore
# https://github.com/tensorflow/tensorflow/issues/34429
./cuda-10.2-no-bin2c-path.patch
];
# On update, it can be useful to steal the changes from gentoo