示例#1
0
 def set_stream(self, stream=None):
     if stream is None:
         stream = cuda.Stream()
     curand.setStream(self._generator, stream.ptr)
示例#2
0
文件： generator.py 项目： 2php/chainer 
 def set_stream(self, stream=None):
     if stream is None:
         stream = cuda.Stream()
     curand.setStream(self._generator, stream.ptr)