aihwkit.simulator.tiles.torch_tile_irdrop_t module

Low level implementation of torch-based tile.

class aihwkit.simulator.tiles.torch_tile_irdrop_t.TorchSimulatorTileIRDropT(x_size, d_size, rpu_config, bias=False)[source]

Bases: TorchSimulatorTile

Torch tile class including time-dependent IR drop calculation.

Parameters:

  • x_size (int) – input size

  • d_size (int) – output size

  • rpu_config (TorchInferenceRPUConfigIRDropT) – resistive processing unit configuration.

  • bias (bool) –

forward(x_input, bias=False, in_trans=False, out_trans=False, is_test=False, non_blocking=False)[source]

General simulator tile forward.

Parameters:

  • x_input (Tensor) –

  • bias (bool) –

  • in_trans (bool) –

  • out_trans (bool) –

  • is_test (bool) –

  • non_blocking (bool) –

Return type:

Tensor