experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Adding a convenience baezel rule - rocm_library

Browse Source
This commit is contained in:
Deven Desai 2020-12-09 23:23:55 +00:00
parent 3f95825a35
commit b65d6c0533
1 changed files with 4 additions and 0 deletions

4
third_party/gpus/rocm/build_defs.bzl.tpl vendored
View File

@ -47,3 +47,7 @@ def if_rocm_is_configured(x):
if rocm_is_configured(): if rocm_is_configured():
return x return x
return [] return []
def rocm_library(copts = [], **kwargs):
"""Wrapper over cc_library which adds default ROCm options."""
native.cc_library(copts = rocm_default_copts() + copts, **kwargs)