mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removed the Triton dependency from the BUILD file

Browse Source 
PiperOrigin-RevId: 597336551
This commit is contained in:
Sergei Lebedev 2024-01-10 13:14:21 -08:00 committed by jax authors
parent 7ad53a6e67
commit 6174145386
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
jaxlib/triton/BUILD
View File

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
load("//jaxlib:jax.bzl", "py_deps", "pybind_extension", "pytype_strict_library")
load("//jaxlib:jax.bzl", "pybind_extension", "pytype_strict_library")
load("@rules_python//python:defs.bzl", "py_library")
load("@llvm-project//mlir:tblgen.bzl", "gentbl_filegroup")
@ -23,11 +23,12 @@ package(
default_visibility = ["//:__subpackages__"],
)
# TODO(slebedev): Make this at least a py_strict_library.
py_library(
name = "compat",
srcs = ["compat.py"],
deps = py_deps("triton"),
deps = [
# Users must add a Triton dependency manually.
],
)
pytype_strict_library(