https://github.com/kevinsala approved this pull request. Except the few nits comments, LGTM. I understand you didn't introduce any big change from the last review (apart from removing the kernel args sizes). https://github.com/llvm/llvm-project/pull/184343