Commit 3e82955
authored
Implement some GB ops (#1342)
* 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 explanation1 parent 02f03c6 commit 3e82955
File tree
6 files changed
+4081
-46
lines changed- diopi_test/python
- configs
- conformance
- impl/torch/functions
- proto/include/diopi
6 files changed
+4081
-46
lines changed
0 commit comments