lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update mpi_broadcast_op.cc 

support uint8 in mpi
This commit is contained in:
Xiang-Li Li 2021-09-25 20:49:29 +08:00 committed by GitHub
parent beebddafd5
commit 6eb1c77474
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
python/jittor/extern/mpi/ops/mpi_broadcast_op.cc vendored
View File

@ -60,6 +60,7 @@ void MpiBroadcastOp::jit_run() {
@if(@strcmp(@Tx,int)==0 || @strcmp(@Tx,int32)==0, MPI_INT)
@if(@strcmp(@Tx,float64)==0 || @strcmp(@Tx,double)==0, MPI_DOUBLE)
@if(@strcmp(@Tx,int64)==0, MPI_DOUBLE_INT)
@if(@strcmp(@Tx,uint8)==0, MPI_CHAR)
)
auto* __restrict__ yp = y->ptr<Tx>();
MPI_Bcast(yp, y->num, T_MPI, root, MPI_COMM_WORLD);