Added official OpenCL headers.

https://github.com/KhronosGroup/OpenCL-Headers

PiperOrigin-RevId: 263040193
This commit is contained in:
A. Unique TensorFlower 2019-08-12 17:30:24 -07:00 committed by TensorFlower Gardener
parent 6202875f3f
commit 4a693b8273
6 changed files with 47 additions and 2 deletions

View File

@ -148,9 +148,9 @@ cc_library(
"//tensorflow/lite/delegates/gpu/common:status",
"//tensorflow/lite/delegates/gpu/common:util",
"//tensorflow/lite/delegates/gpu/gl:portable",
"//third_party/opencl_headers",
"@com_google_absl//absl/types:span",
"@com_google_absl//absl/types:variant",
"@opencl_headers",
],
)

View File

@ -39,7 +39,7 @@ limitations under the License.
#include "absl/types/span.h"
#include "absl/types/variant.h"
#include "third_party/opencl_headers/CL/cl.h"
#include <CL/cl.h>
#include "tensorflow/lite/delegates/gpu/common/data_type.h"
#include "tensorflow/lite/delegates/gpu/common/status.h"
#include "tensorflow/lite/delegates/gpu/common/util.h"

View File

@ -28,6 +28,7 @@ load("//third_party/hwloc:workspace.bzl", hwloc = "repo")
load("//third_party/icu:workspace.bzl", icu = "repo")
load("//third_party/jpeg:workspace.bzl", jpeg = "repo")
load("//third_party/nasm:workspace.bzl", nasm = "repo")
load("//third_party/opencl_headers:workspace.bzl", opencl_headers = "repo")
load("//third_party/kissfft:workspace.bzl", kissfft = "repo")
load("//third_party/keras_applications_archive:workspace.bzl", keras_applications = "repo")
load("//third_party/pasta:workspace.bzl", pasta = "repo")
@ -44,6 +45,7 @@ def initialize_third_party():
kissfft()
jpeg()
nasm()
opencl_headers()
pasta()
# Sanitize a dependency so that it works correctly from code that includes

0
third_party/opencl_headers/BUILD vendored Normal file
View File

28
third_party/opencl_headers/BUILD.bazel vendored Normal file
View File

@ -0,0 +1,28 @@
# Description:
# OpenCL(TM) API Headers
package(default_visibility = ["//visibility:public"])
licenses(["notice"])
exports_files(["LICENSE"])
cc_library(
name = "opencl_headers",
hdrs = [
"CL/cl.h",
"CL/cl_d3d10.h",
"CL/cl_d3d11.h",
"CL/cl_dx9_media_sharing.h",
"CL/cl_egl.h",
"CL/cl_ext.h",
"CL/cl_ext_intel.h",
"CL/cl_gl.h",
"CL/cl_gl_ext.h",
"CL/cl_platform.h",
"CL/cl_va_api_media_sharing_intel.h",
"CL/cl_version.h",
"CL/opencl.h",
],
includes = ["."],
)

View File

@ -0,0 +1,15 @@
"""Loads OpenCL-Headers, used by TF Lite."""
load("//third_party:repo.bzl", "third_party_http_archive")
def repo():
third_party_http_archive(
name = "opencl_headers",
strip_prefix = "OpenCL-Headers-0d5f18c6e7196863bc1557a693f1509adfcee056",
sha256 = "03cbc1fd449399be0422cdb021400f63958ef2c5a7c099a0d8f36a705b312f53",
urls = [
"https://mirror.bazel.build/github.com/KhronosGroup/OpenCL-Headers/archive/0d5f18c6e7196863bc1557a693f1509adfcee056.tar.gz",
"https://github.com/KhronosGroup/OpenCL-Headers/archive/0d5f18c6e7196863bc1557a693f1509adfcee056.tar.gz",
],
build_file = "//third_party/opencl_headers:BUILD.bazel",
)