fc1f6fdf94
This allows JAX to communicate on-device arrays with other libraries, such as PyTorch and CuPy. PiperOrigin-RevId: 290845329 Change-Id: Idd99d81533159bc2ad0c5177b69ac7f30315cb1a
15 lines
288 B
Python
15 lines
288 B
Python
# Description:
|
|
# DLPack is a protocol for sharing arrays between deep learning frameworks.
|
|
|
|
licenses(["notice"]) # Apache 2
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
cc_library(
|
|
name = "dlpack",
|
|
hdrs = [
|
|
"include/dlpack/dlpack.h",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
)
|