Skip to content

Commit

Permalink
[easy] Add NestedTensorMeta to parseDispatchKey (pytorch#94279)
Browse files Browse the repository at this point in the history
ran into this when trying to use `torch.library.Library("aten", "IMPL", "NestedTensorMeta")`

Pull Request resolved: pytorch#94279
Approved by: https://github.com/bdhirsh
  • Loading branch information
mikaylagawarecki authored and pytorchmergebot committed Feb 7, 2023
1 parent 8c835a9 commit 895d478
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions c10/core/DispatchKey.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -311,6 +311,7 @@ c10::DispatchKey parseDispatchKey(const std::string& k) {
{"NestedTensor", c10::DispatchKey::NestedTensor},
{"NestedTensorCPU", c10::DispatchKey::NestedTensorCPU},
{"NestedTensorCUDA", c10::DispatchKey::NestedTensorCUDA},
{"NestedTensorMeta", c10::DispatchKey::NestedTensorMeta},
{"PrivateUse1", c10::DispatchKey::PrivateUse1},
{"PrivateUse2", c10::DispatchKey::PrivateUse2},
{"PrivateUse3", c10::DispatchKey::PrivateUse3},
Expand Down

0 comments on commit 895d478

Please sign in to comment.