Skip to content

Commit

Permalink
pt: add 4 tabulate_fusion op (deepmodeling#3877)
Browse files Browse the repository at this point in the history
**OP**: tabulate_multi_device.cc -> tabulate_fusion_se_a & atten & r & t
**Compile**: add compile file
**Test**: source/tests, se_t has no test, and all other ops pass the
forward and backward tests.

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **New Features**
- Introduced support for tabulating fusion operations for deep learning
models using PyTorch tensors.
- Added functions for forward, backward, and gradient computations for
various fusion operations, handling different data types and device
types.
- Registered new operations with PyTorch libraries for seamless
integration.

- **Chores**
- Updated build configuration to include new source files for fusion
operations.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
  • Loading branch information
3 people authored and Mathieu Taillefumier committed Sep 18, 2024
1 parent 70c5b14 commit 64d5225
Show file tree
Hide file tree
Showing 6 changed files with 7,172 additions and 1 deletion.
2 changes: 1 addition & 1 deletion source/op/pt/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
file(GLOB OP_SRC print_summary.cc comm.cc)
file(GLOB OP_SRC print_summary.cc comm.cc tabulate_multi_device.cc)

add_library(deepmd_op_pt MODULE ${OP_SRC})
# link: libdeepmd libtorch
Expand Down
Loading

0 comments on commit 64d5225

Please sign in to comment.