forked from plaidml/plaidml
-
Notifications
You must be signed in to change notification settings - Fork 0
/
BUILD
99 lines (95 loc) · 2.1 KB
/
BUILD
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# Copyright 2017-2018 Intel Corporation.
load("//bzl:plaidml.bzl", "plaidml_cc_library", "plaidml_cc_test", "plaidml_proto_library")
plaidml_proto_library(
name = "proto",
srcs = [
"opencl.proto",
],
tags = ["opencl"],
visibility = ["//visibility:public"],
deps = [
"//base/context:proto",
"//tile/proto:hal",
],
)
plaidml_cc_library(
name = "opencl",
srcs = [
"buffer.cc",
"buffer.h",
"cl_mem_arena.cc",
"cl_mem_arena.h",
"cl_mem_buffer.cc",
"cl_mem_buffer.h",
"cl_opt.cc",
"cl_opt.h",
"compiler.cc",
"compiler.h",
"compute_kernel.cc",
"compute_kernel.h",
"device.cc",
"device.h",
"device_memory.cc",
"device_memory.h",
"device_set.cc",
"device_set.h",
"device_state.cc",
"device_state.h",
"driver.cc",
"driver.h",
"emitocl.cc",
"event.cc",
"event.h",
"executable.cc",
"executable.h",
"executor.cc",
"executor.h",
"host_memory.cc",
"host_memory.h",
"info.cc",
"kernel.h",
"library.cc",
"library.h",
"ocl.cc",
"ocl.h",
"opencl.cc",
"result.cc",
"result.h",
"shared_memory.cc",
"zero_kernel.cc",
"zero_kernel.h",
],
hdrs = [
"emitocl.h",
"info.h",
],
tags = ["opencl"],
visibility = ["//visibility:public"],
deps = [
":proto_cc",
"//base/shim/opencl",
"//base/util",
"//tile/base",
"//tile/base:hal",
"//tile/hal/util:selector",
"//tile/lang",
"//tile/proto:proto_cc",
"//tile/proto:support",
"@boost//:filesystem",
"@boost//:regex",
],
alwayslink = 1,
)
plaidml_cc_test(
name = "platform_test",
srcs = ["platform_test.cc"],
tags = [
"not_ci",
"opencl",
],
deps = [
":opencl",
"//tile/base:platform_test",
"//tile/platform/local_machine",
],
)