pytorch
dd7e2b7c - [pt2][inductor] update choice caller hashes (#94853)

Commit
2 years ago
[pt2][inductor] update choice caller hashes (#94853) Summary: update the hashing method for `ChoiceCaller` class. `TritonTemplateCaller` objects will now be hashed to: `{name}-({BLOCK_M}, {BLOCK_N}, {BLOCK_K})-{num_stages}-{num_warps}-{code_hash}` for example: `triton_mm-(64, 32, 32)-4-8-cptlntwzcl2gaaofd2oabdwhaqv4ox3lluvbuxitjfhhpz6cyl4o` `ExternKernelCaller` objects will now be hashed to: `{name}-{kwargs.keys()[0]}={kwargs.vals()[0]}-...-{code_hash}` for example: `addmm-alpha=1-beta=1-c4xxd3iocu4yt6z4udrlqnumays7q6mfnfd3qprh4fxgsvyhqdkf` Test Plan: sandcastle Differential Revision: D43285470 Pull Request resolved: https://github.com/pytorch/pytorch/pull/94853 Approved by: https://github.com/jansel, https://github.com/bertmaher
Author
Committer
Parents
Loading