B \F @sddlmZmZddlZddlmZdZdZdZdZ ee Z dZ dZ e eZ d Zed d d d ddddgZddZiZded <ded <ded <ded<dS))divisionprint_functionN) namedtuple@ii _limitsallowed_wave_due_to_sgprallowed_wave_due_to_vgpr allowed_waveallowed_vgpr_per_workitem occupancyreasons suggestionsc Csdd}|dkr|nd}|dkr$|nd}|t}||t}|t}t|}t|}t|t|} |t|t} t} ||kr| d||kr| d| |kr| d|tkr| dd d t | D} | rdn|t} | t }t ||| | || | d S) NcSstt|S)N)intmathZceil)xr6lib/python3.7/site-packages/numba/roc/gcn_occupancy.py_ceilsz#get_limiting_factors.._ceilrr r r group_sizecSsg|] }t|qSr) _suggestions).0rrrr 9sz(get_limiting_factors..)r r r r rrr) simd_per_cu wave_size sgpr_per_simd vgpr_per_simdminmax_wave_countsetaddmax_group_sizesortedmax_inflight_wave_per_cur )rZvgpr_per_workitemZ sgpr_per_waverZworkitem_per_simdZrequired_wave_count_per_simdZrequired_vgpr_per_waver r r r rrZinflight_wave_per_currrrget_limiting_factorss<      r(z;* Cannot allocate enough sGPRs for all resident wavefronts.z;* Cannot allocate enough vGPRs for all resident wavefronts.z?* Launch requires too many wavefronts. Try reducing group-size.z* Exceeds max group size (256).r)Z __future__rrr collectionsrrrZvector_register_file_sizeZ byte_per_VGPRr rr"r'r%r r(rrrrrs0 +