Nigel
/
jittor
forked from jittor/jittor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge branch 'master' of github.com:Jittor/jittor

This commit is contained in:
cxjyxx_me 2021-09-27 23:21:03 -04:00
parent 36faa8b617 4e7dbd327f
commit 9ab20a5536
3 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.py
View File

@ -9,7 +9,7 @@
# file 'LICENSE.txt', which is part of this source code package.
# ***************************************************************
__version__ = '1.2.3.103'
__version__ = '1.2.3.104'
from jittor_utils import lock
with lock.lock_scope():
ori_int = int

1
python/jittor/extern/cuda/nccl/ops/nccl_broadcast_op.cc vendored
View File

@ -47,6 +47,7 @@ void NcclBroadcastOp::jit_run() {
@if(@strcmp(@Tx,int)==0 || @strcmp(@Tx,int32)==0, ncclInt)
@if(@strcmp(@Tx,float64)==0, ncclFloat64)
@if(@strcmp(@Tx,int64)==0, ncclInt64)
@if(@strcmp(@Tx,uint8)==0, ncclUint8)
)
auto* __restrict__ xp = x->ptr<Tx>();
auto* __restrict__ yp = y->ptr<Tx>();

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);