Zhangsheng Lai zunction
lebedov
/ multi_pycuda_ipc_demo.py
Created
September 2, 2013 00:10
Allocate a GPUArray in one process and access it in another process using IPC handles.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python | |
| """ | |
| Allocate a GPUArray in one process and access it in another process using IPC | |
| handles. | |
| """ | |
| import multiprocessing as mp | |
| import numpy as np | |
| import zmq |