Added official OpenCL headers.
https://github.com/KhronosGroup/OpenCL-Headers PiperOrigin-RevId: 263040193
This commit is contained in:
parent
6202875f3f
commit
4a693b8273
@ -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",
|
||||
],
|
||||
)
|
||||
|
||||
|
@ -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"
|
||||
|
@ -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
0
third_party/opencl_headers/BUILD
vendored
Normal file
28
third_party/opencl_headers/BUILD.bazel
vendored
Normal file
28
third_party/opencl_headers/BUILD.bazel
vendored
Normal 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 = ["."],
|
||||
)
|
15
third_party/opencl_headers/workspace.bzl
vendored
Normal file
15
third_party/opencl_headers/workspace.bzl
vendored
Normal 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",
|
||||
)
|
Loading…
Reference in New Issue
Block a user