Visible to Intel only — GUID: GUID-11415167-A7FF-494C-8C5D-FF2E048409B0
Visible to Intel only — GUID: GUID-11415167-A7FF-494C-8C5D-FF2E048409B0
namespace dnnl::sycl_interop
Overview
SYCL interoperability namespace. More…
namespace sycl_interop {
// enums
enum memory_kind;
// global functions
dnnl_sycl_interop_memory_kind_t convert_to_c(memory_kind akind);
engine make_engine(const sycl::device& adevice, const sycl::context& acontext);
sycl::context get_context(const engine& aengine);
sycl::device get_device(const engine& aengine);
stream make_stream(const engine& aengine, sycl::queue& aqueue);
sycl::queue get_queue(const stream& astream);
template <typename T, int ndims = 1>
sycl::buffer<T, ndims> get_buffer(const memory& amemory);
template <typename T, int ndims>
void set_buffer(
memory& amemory,
sycl::buffer<T, ndims>& abuffer
);
memory_kind get_memory_kind(const memory& amemory);
memory make_memory(
const memory::desc& memory_desc,
const engine& aengine,
memory_kind kind,
void* handle = DNNL_MEMORY_ALLOCATE
);
template <typename T, int ndims = 1>
memory make_memory(
const memory::desc& memory_desc,
const engine& aengine,
sycl::buffer<T, ndims>& abuffer
);
sycl::event execute(
const dnnl::primitive& aprimitive,
const stream& astream,
const std::unordered_map<int, memory>& args,
const std::vector<sycl::event>& deps = {}
);
} // namespace sycl_interop
Detailed Documentation
SYCL interoperability namespace.
Global Functions
dnnl_sycl_interop_memory_kind_t convert_to_c(memory_kind akind)
Converts a memory allocation kind enum value from C++ API to C API type.
Parameters:
akind |
C++ API memory allocation kind enum value. |
Returns:
Corresponding C API memory allocation kind enum value.
engine make_engine(const sycl::device& adevice, const sycl::context& acontext)
Constructs an engine from SYCL device and context objects.
Parameters:
adevice |
SYCL device. |
acontext |
SYCL context. |
Returns:
Created engine.
sycl::context get_context(const engine& aengine)
Returns the SYCL context associated with an engine.
Parameters:
aengine |
Engine to query. |
Returns:
The underlying SYCL device of the engine.
sycl::device get_device(const engine& aengine)
Returns the SYCL device associated with an engine.
Parameters:
aengine |
Engine to query. |
Returns:
The underlying SYCL context of the engine.
stream make_stream(const engine& aengine, sycl::queue& aqueue)
Creates an execution stream for a given engine associated with a SYCL queue.
Parameters:
aengine |
Engine object to use for the stream. |
aqueue |
SYCL queue to use for the stream. |
Returns:
An execution stream.
sycl::queue get_queue(const stream& astream)
Returns the SYCL queue associated with an execution stream.
Parameters:
astream |
Execution stream to query. |
Returns:
SYCL queue object.
template <typename T, int ndims = 1>
sycl::buffer<T, ndims> get_buffer(const memory& amemory)
Returns the SYCL buffer associated with a memory object.
Throws an exception if the memory allocation kind associated with the memory object is not equal to dnnl::sycl_interop::memory_kind::buffer.
Parameters:
T |
Type of the requested buffer. |
ndims |
Number of dimensions of the requested buffer. |
amemory |
Memory object. |
Returns:
SYCL buffer associated with the memory object.
template <typename T, int ndims>
void set_buffer(
memory& amemory,
sycl::buffer<T, ndims>& abuffer
)
Sets SYCL buffer associated with a memory object.
Parameters:
T |
Type of the buffer. |
ndims |
Number of dimensions of the buffer. |
amemory |
Memory object to change. |
abuffer |
SYCL buffer. |
memory_kind get_memory_kind(const memory& amemory)
Returns the memory allocation kind associated with a memory object.
Parameters:
amemory |
A memory object. |
Returns:
The underlying memory allocation kind of the memory object.
memory make_memory(
const memory::desc& memory_desc,
const engine& aengine,
memory_kind kind,
void* handle = DNNL_MEMORY_ALLOCATE
)
Creates a memory object.
Unless handle is equal to DNNL_MEMORY_NONE or DNNL_MEMORY_ALLOCATE, the constructed memory object will have the underlying buffer set. In this case, the buffer will be initialized as if:
dnnl::memory::set_data_handle() had been called, if memory_kind is equal to dnnl::sycl_interop::memory_kind::usm, or
dnnl::sycl_interop::set_buffer() has been called, if memory_kind is equal to dnnl::sycl_interop::memory_kind::buffer.
Parameters:
memory_desc |
Memory descriptor. |
aengine |
Engine to use. |
kind |
Memory allocation kind to specify the type of handle. |
handle |
Handle of the memory buffer to use as an underlying storage.
|
Returns:
Created memory object.
template <typename T, int ndims = 1>
memory make_memory(
const memory::desc& memory_desc,
const engine& aengine,
sycl::buffer<T, ndims>& abuffer
)
Constructs a memory object from a SYCL buffer.
Parameters:
memory_desc |
Memory descriptor. |
aengine |
Engine to use. |
abuffer |
A SYCL buffer to use. |
Returns:
Created memory object.
sycl::event execute(
const dnnl::primitive& aprimitive,
const stream& astream,
const std::unordered_map<int, memory>& args,
const std::vector<sycl::event>& deps = {}
)
Executes computations specified by the primitive in a specified stream and returns a SYCL event.
Arguments are passed via an arguments map containing <index, memory object> pairs. The index must be one of the DNNL_ARG_* values such as DNNL_ARG_SRC, and the memory must have a memory descriptor matching the one returned by dnnl::primitive_desc::query_md (query::exec_arg_md, index) unless using dynamic shapes (see DNNL_RUNTIME_DIM_VAL).
Parameters:
aprimitive |
Primitive to execute. |
astream |
Stream object. The stream must belong to the same engine as the primitive. |
args |
Arguments map. |
deps |
Optional vector with sycl::event dependencies. |
Returns:
Output event.