def test_sequential_move_to_cpu(self):
"""Test moving AnalogSequential to CPU (from CPU)."""
layer = self.get_layer()
# Create a container and move to cuda.
model = AnalogSequential(layer)
model.cpu()
# Assert the tile is still on CPU.
self.assertIsInstance(layer.analog_tile, AnalogTile)
def test_sequential_move_to_cpu(self):
"""Test moving AnalogSequential to CPU (from CPU)."""
layer = self.get_layer()
# Create a container and move to cuda.
model = AnalogSequential(layer)
model.cpu()
analog_tile = layer.analog_tile
self.assertEqual(analog_tile.device, device('cpu'))
self.assertEqual(analog_tile.get_analog_ctx().data.device,
device('cpu'))
if analog_tile.shared_weights is not None:
self.assertEqual(analog_tile.shared_weights.data.device,
device('cpu'))
self.assertEqual(analog_tile.shared_weights.data.size()[0],
analog_tile.tile.get_d_size())
self.assertEqual(analog_tile.shared_weights.data.size()[1],
analog_tile.tile.get_x_size())
# Assert the tile is still on CPU.
self.assertIsInstance(layer.analog_tile.tile, tiles.AnalogTile)