[AutoWS] Move p computation closer to use#564

njriasan wants to merge 2 commits intometa-pytorch:mainmeta-pytorch/tritonbench:mainfrom

njriasan:njriasan/tmp_changenjriasan/tritonbench:njriasan/tmp_changeCopy head branch name to clipboard