22e07fb204
We have a use after free caused by memory coruption, a segmentation fault caused by memory corruption, several memory leaks and an undefined behavior when taking the reference of a nullptr. PiperOrigin-RevId: 332568894 Change-Id: Ife0fc05e103b35325094ae5d822ee5fdea764572
29 lines
644 B
Python
29 lines
644 B
Python
load("//tensorflow:tensorflow.bzl", "cuda_py_test")
|
|
|
|
package(
|
|
default_visibility = ["//visibility:private"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
py_library(
|
|
name = "dlpack",
|
|
srcs = ["dlpack.py"],
|
|
srcs_version = "PY2AND3",
|
|
visibility = ["//tensorflow:__subpackages__"],
|
|
deps = [
|
|
"//tensorflow/python:pywrap_tensorflow",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "dlpack_test",
|
|
srcs = ["dlpack_test.py"],
|
|
srcs_version = "PY2AND3",
|
|
deps = [
|
|
":dlpack",
|
|
"//tensorflow/python/eager:test",
|
|
"@absl_py//absl/testing:absltest",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|