ó ßüÚ\c@shddlmZmZddlZejdƒZejdƒZd„Zd„Zd„Z d„Z dS( iÿÿÿÿ(t serializet deserializeNtnumpyttorchcCswtjdƒ}tj|ƒ}t|ƒ\}}|ddksFt‚t||ƒ}||jƒkjƒsst‚dS(Ni t serializertdask( tnptarangeRtTensorRtAssertionErrorRRtall(txtttheadertframestt2((sDlib/python2.7/site-packages/distributed/protocol/tests/test_torch.pyt test_tensors cCs‰tjdƒ}tj|ƒ}tj|ƒd|_tt|ƒŒ}|jƒ|kj ƒsdt ‚|jjƒdkj ƒs…t ‚dS(Ni i( RRRRt zeros_liketgradRRRR R (R R R((sDlib/python2.7/site-packages/distributed/protocol/tests/test_torch.pyt test_grads cCsdtjdƒ}|jjjƒ}t|ƒ\}}t||ƒ}t|ƒt|ƒks`t‚dS(Nt torchvision( tpytestt importorskiptmodelstresnettresnet18RRtstrR (RtmodelR Rtmodel2((sDlib/python2.7/site-packages/distributed/protocol/tests/test_torch.pyt test_resnets cCs|tjjddƒ}tj|dtdtjƒ}tt|ƒŒ}|j sTt ‚tj ||j ƒj ƒƒsxt ‚dS(Niit requires_gradtdtype(RtrandomtrandRttensortTruetfloatRRRR tallclosetdetach_R(taR R((sDlib/python2.7/site-packages/distributed/protocol/tests/test_torch.pyttest_deserialize_grad$s ( tdistributed.protocolRRRRRRRRRR((((sDlib/python2.7/site-packages/distributed/protocol/tests/test_torch.pyts