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

avoid references to fiel out of directory...

avoid references to fiel out of directory request for packaging, it would be nice for example not to reference files outside of exampel directory copy ../../_static/logo.png in logo/logo.png use subfolder for demo purpose of targetting subfolder in demo notebook

Thomas Kluyver - - Load All Authors

File last commit:

r6455:15863dc1


                r9992:713f1db0

Download file

             noncopying.py
        
                    44 lines
            
             | 866 B
            
                | text/x-python
            
             |
                PythonLexer
            
             / docs / 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())