Skip to content

std1.0.0-beta-7 SafeC docs

All items

Module

ml
ml::device

Structs

DeviceStats
DevArgs
DevGemm
DevConv
DeviceOps

Type aliases

DevAvailFn
DevAllocFn
DevReleaseFn
DevUploadFn
DevDownloadFn
DevCopyFn
DevSyncFn
DevGemmFn
DevGemmBatchedFn
DevConvFn
DevStorageFn
DevLaunchFn
DevOffsetFn

Functions

device_compiled
device_available
device_auto
device_set_default
device_default
device_name
device_sync
device_bytes_reserved
device_cache_clear
device_stats
device_stats_reset
device_set_precision
device_precision
device_set_precision_pick
device_precision_pick
device_loss_scale

Macros

DEVICE_CPU
DEVICE_ROCM
DEVICE_CUDA
DEVICE_MPS
DEVICE_VULKAN
DEVICE_WEBGPU
DEVICE_COUNT
DEVICE_AUTO
DEVK_PTRS
DEVK_INTS
DEVK_FLOATS
DEVICE_FP32
DEVICE_FP16
DEVICE_BF16
DEVICE_PRECISION_KEEP
DEV_TAG_SHIFT
DEV_TAG_MASK
DEV_CONV_FWD
DEV_CONV_BWD_DATA
DEV_CONV_BWD_WEIGHTS
DEVICE_PICK_TIMED
DEVICE_PICK_ALWAYS

std :: ml::device

Macro DEVICE_ROCM

#include <std/ml/device.h> source

#define DEVICE_ROCM 1

On this page