Skip to content

Instantly share code, notes, and snippets.

@zou3519

zou3519/fallthrough.cpp Secret

Created June 30, 2023 21:54
Show Gist options
  • Save zou3519/d09a5f4b1afe2430af09fea67c6ff2c8 to your computer and use it in GitHub Desktop.
Save zou3519/d09a5f4b1afe2430af09fea67c6ff2c8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
fallthrough.cpp
TORCH_LIBRARY_IMPL(your_library, Autograd) {
m.impl("your_op_name", torch::CppFunction::makeFallthrough());
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment