Skip to content

[Torch] Implements __torch_func__ protocol#400

Merged
yaoyaoding merged 1 commit intohidet-org:mainfrom
yaoyaoding:torch-func-protocol
Dec 29, 2023
Merged

[Torch] Implements __torch_func__ protocol#400
yaoyaoding merged 1 commit intohidet-org:mainfrom
yaoyaoding:torch-func-protocol

Conversation

@yaoyaoding
Copy link
Copy Markdown
Member

This PR implements the __torch_func__ protocol as suggested at https://pytorch.org/docs/stable/notes/extending.html#extending-torch-with-a-tensor-like-type

@yaoyaoding yaoyaoding merged commit 57154a3 into hidet-org:main Dec 29, 2023
@yaoyaoding yaoyaoding deleted the torch-func-protocol branch December 29, 2023 19:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant