python.pkgs.tensorflow: fix build with cudatoolkit 10.2

This commit is contained in:
Pavol Rusnak 2020-01-04 17:34:29 +01:00
parent e7e5fc9584
commit 9a982d3540
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
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"; sha256 = "077cpj0kzyqxzdya1dwh8df17zfzhqn7c685hx6iskvw2979zg2n";
}) })
./lift-gast-restriction.patch ./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 # On update, it can be useful to steal the changes from gentoo