diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2015-11-12 11:27:00 -0800 |
|---|---|---|
| committer | Vijay Vasudevan <vrv@google.com> | 2015-11-12 11:27:00 -0800 |
| commit | 4dffee7f62d81ec9173aba1b0ef6b96e47f8037c (patch) | |
| tree | 4b5c04b37afe45fdc5f1729252514a2770fbf1ab /third_party/gpus/crosstool/clang | |
| parent | f2102f4e2c1c87f1d1bf9ab856a2849c54478760 (diff) | |
TensorFlow: Minor updates to docs, BUILD, GPU config / perf, etc.
Changes:
- Updates to op documentation and index by Josh
- More changes to BUILD files for python 3 support by @girving
- Fix to Eigen to use DenseIndex everywhere by @jiayq
- Enable configuration for cuda compute capability by @zheng-xq,
including updates to docs.
- Route aggregation method through optimizer by schuster
- Updates to install instructions for bazel 0.1.1.
Base CL: 107702099
Diffstat (limited to 'third_party/gpus/crosstool/clang')
| -rwxr-xr-x | third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc b/third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc index 0f419a8b61..cdbd1823ea 100755 --- a/third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc +++ b/third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc @@ -232,9 +232,14 @@ def InvokeNvcc(argv, log=False): srcs = ' '.join(src_files) out = ' -o ' + out_file[0] - nvccopts = ' '.join([ - r'-gencode=arch=compute_35,\"code=sm_35,compute_35\"', - r'-gencode=arch=compute_52,\"code=sm_52,compute_52\"',]) + # "configure" uses the specific format to substitute the following string. + # If you change it, make sure you modify "configure" as well. + supported_cuda_compute_capabilities = [ "3.5", "5.2" ] + nvccopts = '' + for capability in supported_cuda_compute_capabilities: + capability = capability.replace('.', '') + nvccopts += r'-gencode=arch=compute_%s,\"code=sm_%s,compute_%s\" ' % ( + capability, capability, capability) nvccopts += ' ' + nvcc_compiler_options nvccopts += undefines nvccopts += defines @@ -260,8 +265,8 @@ def InvokeNvcc(argv, log=False): ' -I .' + ' -x cu ' + opt + includes + ' -c ' + srcs + out) - # TODO(zhengxq): for some reason, 'gcc' needs this help to find 'as'. - # Need to investigate and fix. + # TODO(zhengxq): for some reason, 'gcc' needs this help to find 'as'. + # Need to investigate and fix. cmd = 'PATH=' + PREFIX_DIR + ' ' + cmd if log: Log(cmd) return os.system(cmd) |