Correct HIP_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK query
We should be returning the max workgroup size calculated by the compiler. Change-Id: If86590efbb9b291f470bdbe87e5df992e661c539
Tá an tiomantas seo le fáil i:
tiomanta ag
Vladislav Sytchenko
tuismitheoir
276bfc9667
tiomantas
b5f9d2f818
@@ -317,8 +317,7 @@ hipError_t hipFuncGetAttribute(int* value, hipFunction_attribute attrib, hipFunc
|
|||||||
*value = static_cast<int>(wrkGrpInfo->localMemSize_);
|
*value = static_cast<int>(wrkGrpInfo->localMemSize_);
|
||||||
break;
|
break;
|
||||||
case HIP_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK:
|
case HIP_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK:
|
||||||
*value = static_cast<int>(wrkGrpInfo->wavefrontPerSIMD_
|
*value = static_cast<int>(wrkGrpInfo->size_);
|
||||||
* wrkGrpInfo->wavefrontSize_);
|
|
||||||
break;
|
break;
|
||||||
case HIP_FUNC_ATTRIBUTE_CONST_SIZE_BYTES:
|
case HIP_FUNC_ATTRIBUTE_CONST_SIZE_BYTES:
|
||||||
*value = 0;
|
*value = 0;
|
||||||
|
|||||||
@@ -313,7 +313,7 @@ bool ihipGetFuncAttributes(const char* func_name, amd::Program* program, hipFunc
|
|||||||
func_attr->localSizeBytes = wginfo->privateMemSize_;
|
func_attr->localSizeBytes = wginfo->privateMemSize_;
|
||||||
func_attr->sharedSizeBytes = wginfo->localMemSize_;
|
func_attr->sharedSizeBytes = wginfo->localMemSize_;
|
||||||
func_attr->maxDynamicSharedSizeBytes = wginfo->availableLDSSize_ - wginfo->localMemSize_;
|
func_attr->maxDynamicSharedSizeBytes = wginfo->availableLDSSize_ - wginfo->localMemSize_;
|
||||||
func_attr->maxThreadsPerBlock = wginfo->wavefrontSize_;
|
func_attr->maxThreadsPerBlock = wginfo->size_;
|
||||||
func_attr->numRegs = wginfo->usedVGPRs_;
|
func_attr->numRegs = wginfo->usedVGPRs_;
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
|
|||||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir