Skip to content

Commit

Permalink
Implement some GB ops (#1342)
Browse files Browse the repository at this point in the history
* add expm1 on diopi torch impl

* add tan on diopi torch impl

* add acos on diopi torch impl

* fix format & add sinh on diopi torch impl

* add cosh on diopi torch impl

* add asinh on diopi torch impl

* add acosh on diopi torch impl

* add atanh on diopi torch impl

* conbine some hyperbolic trigo fucntion diopi test into hyperbolic_trigo_function

* add argmin on diopi torch impl & del diopi test hyperbolic_trigo_function's save_args param

* add argsort on diopi torch impl

* add sort_backward on diopi torch impl

* refactor diopi_test for sort_backward & add cumsum_backward

* add complex on diopi torch impl

* add conj on diopi torch impl

* refactor some format & add imag on diopi torch impl

* add real on diopi torch impl

* fix setCurStream & add grid_sample & prepare for diopiPool2d

* add norm_backward, normalize, normalize_backward, layer_normGB, layer_normGB_backward

* prepare for pool1d, pool2d, pool3d

* finish diopi_pool related

* add part of pool1d in diopi_configs.py and diopi_functions.py

* finish all of pool1d, pool2d and pool3d

* update layer_normGB, add instance_norm_backward

* add conv_transpose3d

* fix clang format

* remove conflicted layer_norm

* finish comment and fix fused_adam caused by previously merge

* remove unused code & try to fix adam

* add layerNorm GB national standard operator's explanation
  • Loading branch information
DoorKickers authored Oct 18, 2024
1 parent 02f03c6 commit 3e82955
Show file tree
Hide file tree
Showing 6 changed files with 4,081 additions and 46 deletions.
Loading

0 comments on commit 3e82955

Please sign in to comment.