AI/rocm-systems
2
0
Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

SWDEV-310187 - Change flag to keep track of aligned sizes instead of expanded patterns.

Telusuri Sumber 
Change-Id: I763feda8688bb1b7b11033a2a8cba0f69f07167d
This commit is contained in:
kjayapra-amd
2021-11-18 14:55:31 -05:00
committed by Karthik Jayaprakash
orang tua b192beea52
melakukan 8307886644
3 mengubah file dengan 12 tambahan dan 11 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
rocclr/device/rocm/rocblit.cpp
+8 -7
Melihat File
@@ -1966,17 +1966,18 @@ bool KernelBlitManager::fillBuffer(device::Memory& memory, const void* pattern,
FillBufferInfo::PackInfo(memory, size[0], origin[0], pattern, patternSize, packed_vector);
size_t overall_offset = origin[0];
uint fillType = FillBufferAligned;
for (auto& packed_obj: packed_vector) {
uint fillType = FillBufferAligned;
size_t globalWorkOffset[3] = {0, 0, 0};
size_t globalWorkSize = amd::alignUp(packed_obj.fill_size_, 256);
size_t localWorkSize = 256;
uint32_t kpattern_size32 = (packed_obj.pattern_expanded_) ? sizeof(size_t) : patternSize;
uint32_t kpattern_size32 = (packed_obj.pattern_aligned_) ? sizeof(size_t) : patternSize;
size_t kfill_size = packed_obj.fill_size_/kpattern_size32;
size_t koffset = overall_offset;
overall_offset += packed_obj.fill_size_;
size_t globalWorkOffset[3] = {0, 0, 0};
size_t globalWorkSize = amd::alignUp(kfill_size, 256);
size_t localWorkSize = 256;
uint32_t alignment = (kpattern_size32 & 0x7) == 0 ?
sizeof(uint64_t) :
(kpattern_size32 & 0x3) == 0 ?
@@ -2018,7 +2019,7 @@ bool KernelBlitManager::fillBuffer(device::Memory& memory, const void* pattern,
auto constBuf = reinterpret_cast<address>(constantBuffer_->getHostMem()) + constBufOffset;
// If pattern has been expanded, use the expanded pattern, otherwise use the default pattern.
if (packed_obj.pattern_expanded_) {
if (packed_obj.pattern_aligned_) {
memcpy(constBuf, &packed_obj.expanded_pattern_, kpattern_size32);
} else {
memcpy(constBuf, pattern, kpattern_size32);
@@ -2027,8 +2028,8 @@ bool KernelBlitManager::fillBuffer(device::Memory& memory, const void* pattern,
mem = as_cl<amd::Memory>(gpuCB->owner());
setArgument(kernels_[fillType], 4, sizeof(cl_mem), &mem, constBufOffset);
koffset /= alignment;
kpattern_size32 /= alignment;
koffset /= alignment;
setArgument(kernels_[fillType], 5, sizeof(uint32_t), &kpattern_size32);
setArgument(kernels_[fillType], 6, sizeof(koffset), &koffset);