Jimmy MA

Results 19 comments of Jimmy MA

spu 支持的是可 jit 的 jax 函数,可参考 log 里的文档 https://jax.readthedocs.io/en/latest/errors.html#jax.errors.TracerBoolConversionError 对你的程序进行修改,让其支持 jax 的 jit。看起来是一个控制流相关的问题。

There are various granularities of parallelism and vectorization within the SPU. It's not clear what your specific requirements are. How much data do you need to process and what granularity...

你只需要让 P0, P1, P2 三方同步 seed 然后产生随机数即可

可以参考现有的实现,在初始化时同步即可

可以用 yacl 里的 hash 算法,https://github.com/secretflow/yacl/blob/main/yacl/crypto/hash/hash_interface.h

P0 调 r0=genCommonRandom({0,1}),block 了在从 recv from P1,同理 P1 block 了在 recv from P2,P2 block 了在 recv from P0,这不就死锁了吗?

检查下 seed 和 counter 是否符合预期

这个是访问 dockerhub 的网络问题,不是镜像本身的问题