#NEWLINE# @ attr . gpu #NEWLINE# def test_pickle_gpu( self ) : #NEWLINE# #INDENT# self . fs2 . to_gpu ( ) #NEWLINE# fs2_serialized = pickle . dumps ( fs2_loaded . fs2 ) #NEWLINE# fs2_loaded = pickle . loads ( fs2_serialized ) #NEWLINE# fs2_loaded . to_cpu ( ) #NEWLINE# self . fs2 . to_cpu ( ) #NEWLINE# self . assertTrue ( ( self . fs2 . b . p . data == fs2_loaded . b . p . data ) . all ( ) ) #NEWLINE# self . assertTrue ( ( self . fs2 . fs1 . a . p . data == fs2_loaded . fs1 . a . p . data ) . all ( ) )