Skip to content

Commit

Permalink
feat(pir): reg barrier
Browse files Browse the repository at this point in the history
  • Loading branch information
xiaoyewww committed Mar 18, 2024
1 parent b9136e4 commit 590343b
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 3 deletions.
1 change: 0 additions & 1 deletion paddle/fluid/pir/dialect/op_generator/ops_api_gen.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,6 @@
'all_reduce',
'all_reduce_',
'barrier',
'barrier_',
'c_allgather',
'c_allreduce_avg',
'c_allreduce_avg_',
Expand Down
2 changes: 0 additions & 2 deletions paddle/fluid/pir/dialect/operator/ir/ops.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -138,8 +138,6 @@
output : Tensor(out)
kernel :
func : barrier
data_type : x
inplace : (x -> out)

- op : batch_norm
args : (Tensor x, Tensor mean, Tensor variance, Tensor scale, Tensor bias, bool is_test, float momentum, float epsilon, str data_format, bool use_global_stats, bool trainable_statistics)
Expand Down

0 comments on commit 590343b

Please sign in to comment.