Searched refs:denseStrides (Results 1 – 2 of 2) sorted by relevance
164 llvm::SmallVector<int64_t, 4> denseStrides(rank); in mgpuMemHostRegisterMemRef() local168 std::partial_sum(sizes.rbegin(), sizes.rend(), denseStrides.rbegin(), in mgpuMemHostRegisterMemRef()170 auto sizeBytes = denseStrides.front() * elementSizeBytes; in mgpuMemHostRegisterMemRef()173 std::rotate(denseStrides.begin(), denseStrides.begin() + 1, in mgpuMemHostRegisterMemRef()174 denseStrides.end()); in mgpuMemHostRegisterMemRef()175 denseStrides.back() = 1; in mgpuMemHostRegisterMemRef()176 assert(strides == llvm::makeArrayRef(denseStrides)); in mgpuMemHostRegisterMemRef()
178 int64_t *denseStrides = (int64_t *)alloca(rank * sizeof(int64_t)); in mgpuMemHostRegisterMemRef() local181 denseStrides[i] = runningStride; in mgpuMemHostRegisterMemRef()184 uint64_t sizeBytes = sizes[0] * denseStrides[0] * elementSizeBytes; in mgpuMemHostRegisterMemRef()188 assert(strides[i] == denseStrides[i] && in mgpuMemHostRegisterMemRef()