@@ -695,7 +695,7 @@ RunGivenTwoCommandQueuesHavingCommandListsWithScratchSpaceThenSuccessIsReturnedT
695
695
uint32_t expectedMemorySize = (arraySize * 2 + 1 ) * typeSize - 4 ;
696
696
uint32_t inMemorySize = expectedMemorySize;
697
697
uint32_t outMemorySize = expectedMemorySize;
698
- uint32_t offsetMemorySize = 128 * arraySize ;
698
+ uint32_t offsetMemorySize = 1281 * typeSize ;
699
699
700
700
ze_device_mem_alloc_desc_t deviceDesc = {};
701
701
deviceDesc.stype = ZE_STRUCTURE_TYPE_DEVICE_MEM_ALLOC_DESC;
@@ -759,9 +759,9 @@ RunGivenTwoCommandQueuesHavingCommandListsWithScratchSpaceThenSuccessIsReturnedT
759
759
groupSizeY, groupSizeZ);
760
760
size_t groupSize = groupSizeX * groupSizeY * groupSizeZ;
761
761
lzt::set_group_size (scratch_function, groupSizeX, groupSizeY, groupSizeZ);
762
- lzt::set_argument_value (scratch_function, 2 , sizeof (dstBuffer ), &dstBuffer );
763
- lzt::set_argument_value (scratch_function, 1 , sizeof (srcBuffer ), &srcBuffer );
764
- lzt::set_argument_value (scratch_function, 0 , sizeof (offsetBuffer),
762
+ lzt::set_argument_value (scratch_function, 0 , sizeof (srcBuffer ), &srcBuffer );
763
+ lzt::set_argument_value (scratch_function, 1 , sizeof (dstBuffer ), &dstBuffer );
764
+ lzt::set_argument_value (scratch_function, 2 , sizeof (offsetBuffer),
765
765
&offsetBuffer);
766
766
// if groupSize is greater then memory count, then at least one thread group
767
767
// should be dispatched
0 commit comments