forked from plaidml/plaidml
-
Notifications
You must be signed in to change notification settings - Fork 0
/
executable.h
32 lines (24 loc) · 858 Bytes
/
executable.h
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
// Copyright 2018, Intel Corporation.
#pragma once
#include <memory>
#include <vector>
#include "tile/base/hal.h"
#include "tile/hal/opencl/kernel.h"
namespace vertexai {
namespace tile {
namespace hal {
namespace opencl {
class Executable final : public hal::Executable {
public:
explicit Executable(std::vector<std::unique_ptr<Kernel>> kernels);
std::shared_ptr<hal::Event> Run(const context::Context& ctx, std::size_t kernel_index,
const std::vector<std::shared_ptr<hal::Buffer>>& params,
const std::vector<std::shared_ptr<hal::Event>>& dependencies,
bool enable_profiling = false) final;
private:
std::vector<std::unique_ptr<Kernel>> kernels_;
};
} // namespace opencl
} // namespace hal
} // namespace tile
} // namespace vertexai