Home
last modified time | relevance | path

Searched refs:elementSizeBytes (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/ExecutionEngine/
H A DCudaRuntimeWrappers.cpp176 int64_t elementSizeBytes) { in mgpuMemHostRegisterMemRef() argument
184 uint64_t sizeBytes = sizes[0] * denseStrides[0] * elementSizeBytes; in mgpuMemHostRegisterMemRef()
191 auto *ptr = descriptor->data + descriptor->offset * elementSizeBytes; in mgpuMemHostRegisterMemRef()
H A DRocmRuntimeWrappers.cpp162 int64_t elementSizeBytes) { in mgpuMemHostRegisterMemRef() argument
170 auto sizeBytes = denseStrides.front() * elementSizeBytes; in mgpuMemHostRegisterMemRef()
178 auto ptr = descriptor->data + descriptor->offset * elementSizeBytes; in mgpuMemHostRegisterMemRef()