upstream/ipython Files · examples/parallel/demo/noncopying.py

Merge pull request from minrk/even-more-ways-to-get-ifaces...

Merge pull request from minrk/even-more-ways-to-get-ifaces Add even more ways to populate localinterfaces use netifaces for faster IPython.utils.localinterfaces when availlable, Parse subprocess output from ifconfig / ip addr / ipconfig. Lower priority than netifaces, but still higher priority than socket.gethostbyname. Fallback to gethostname otherwise. Should be much faster in worst case scenario where machine are badly configurred and can wait up to ~30s to start ipython. Slighly slower in other cases.

Brian Granger - - Load All Authors

File last commit:

r9190:20a102a5


                r12911:aeeb7f5a

Download file

             noncopying.py
        
                    44 lines
            
             | 866 B
            
                | text/x-python
            
             |
                PythonLexer
            
             / examples / parallel / demo / noncopying.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      """non-copying sends"""

      import zmq

      import numpy

      n = 10

      iface = 'inproc://pub'

      ctx = zmq.Context()

      p = ctx.socket(zmq.PUB)

      p.bind(iface)

      # connect 2 subs

      s1 = ctx.socket(zmq.SUB)

      s1.connect(iface)

      s1.setsockopt(zmq.SUBSCRIBE, '')

      s2 = ctx.socket(zmq.SUB)

      s2.connect(iface)

      s2.setsockopt(zmq.SUBSCRIBE, '')

      A = numpy.random.random((1024,1024))

      # send

      p.send(A, copy=False)

      # recv on 1 non-copy

      msg1 = s1.recv(copy=False)

      B1 = numpy.frombuffer(msg1.buffer, dtype=A.dtype).reshape(A.shape)

      # recv on 2 copy

      msg2 = s2.recv(copy=False)

      B2 = numpy.frombuffer(buffer(msg2.bytes), dtype=A.dtype).reshape(A.shape)

      print((B1==B2).all())

      print((B1==A).all())

      A[0][0] += 10

      print("~")

      # after changing A in-place, B1 changes too, proving non-copying sends

      print((B1==A).all())

      # but B2 is fixed, since it called the msg.bytes attr, which copies

      print((B1==B2).all())

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				"""non-copying sends"""
				import zmq
				import numpy

				n = 10
				iface = 'inproc://pub'

				ctx = zmq.Context()

				p = ctx.socket(zmq.PUB)
				p.bind(iface)

				# connect 2 subs
				s1 = ctx.socket(zmq.SUB)
				s1.connect(iface)
				s1.setsockopt(zmq.SUBSCRIBE, '')

				s2 = ctx.socket(zmq.SUB)
				s2.connect(iface)
				s2.setsockopt(zmq.SUBSCRIBE, '')

				A = numpy.random.random((1024,1024))

				# send
				p.send(A, copy=False)
				# recv on 1 non-copy
				msg1 = s1.recv(copy=False)
				B1 = numpy.frombuffer(msg1.buffer, dtype=A.dtype).reshape(A.shape)
				# recv on 2 copy
				msg2 = s2.recv(copy=False)
				B2 = numpy.frombuffer(buffer(msg2.bytes), dtype=A.dtype).reshape(A.shape)

				print((B1==B2).all())
				print((B1==A).all())
				A[0][0] += 10
				print("~")
				# after changing A in-place, B1 changes too, proving non-copying sends
				print((B1==A).all())
				# but B2 is fixed, since it called the msg.bytes attr, which copies
				print((B1==B2).all())