mxnet
Functions
dnnl_sycl.h File Reference
#include "oneapi/dnnl/dnnl.h"
#include "oneapi/dnnl/dnnl_sycl_types.h"
Include dependency graph for dnnl_sycl.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions

dnnl_status_t DNNL_API dnnl_sycl_interop_engine_create (dnnl_engine_t *engine, const void *device, const void *context)
 
dnnl_status_t DNNL_API dnnl_sycl_interop_engine_get_context (dnnl_engine_t engine, void **context)
 
dnnl_status_t DNNL_API dnnl_sycl_interop_engine_get_device (dnnl_engine_t engine, void **device)
 
dnnl_status_t DNNL_API dnnl_sycl_interop_memory_create (dnnl_memory_t *memory, const dnnl_memory_desc_t *memory_desc, dnnl_engine_t engine, dnnl_sycl_interop_memory_kind_t memory_kind, void *handle)
 
dnnl_status_t DNNL_API dnnl_sycl_interop_memory_get_memory_kind (const_dnnl_memory_t memory, dnnl_sycl_interop_memory_kind_t *memory_kind)
 
dnnl_status_t DNNL_API dnnl_sycl_interop_memory_set_buffer (dnnl_memory_t memory, void *buffer, dnnl_stream_t stream)
 
dnnl_status_t DNNL_API dnnl_sycl_interop_stream_create (dnnl_stream_t *stream, dnnl_engine_t engine, void *queue)
 
dnnl_status_t DNNL_API dnnl_sycl_interop_stream_get_queue (dnnl_stream_t stream, void **queue)
 
dnnl_status_t DNNL_API dnnl_sycl_interop_primitive_execute (const_dnnl_primitive_t primitive, dnnl_stream_t stream, int nargs, const dnnl_exec_arg_t *args, const void *deps, void *return_event)