提交 12586a61 authored 作者: Matthew Rocklin's avatar Matthew Rocklin

mpi_cmp -> mpi_cmps in test_mpi

上级 3aaa820f
from theano.tensor.io import send, recv, mpi_cmp, MPISend, MPISendWait
from theano.tensor.io import send, recv, mpi_cmps, MPISend, MPISendWait
import theano
import subprocess
import os
from theano.gof.sched import sort_schedule_fn
mpi_scheduler = sort_schedule_fn(mpi_cmp)
mpi_scheduler = sort_schedule_fn(*mpi_cmps)
mpi_linker = theano.OpWiseCLinker(schedule=mpi_scheduler)
mpi_mode = theano.Mode(linker=mpi_linker)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论