(self, *args, **kwargs)
| 56 | self.state = 1 |
| 57 | |
| 58 | def forward(self, *args, **kwargs): |
| 59 | if ( |
| 60 | self.onload_dtype == self.computation_dtype |
| 61 | and self.onload_device == self.computation_device |
| 62 | ): |
| 63 | module = self.module |
| 64 | else: |
| 65 | module = copy.deepcopy(self.module).to( |
| 66 | dtype=self.computation_dtype, device=self.computation_device |
| 67 | ) |
| 68 | return module(*args, **kwargs) |
| 69 | |
| 70 | |
| 71 |
nothing calls this directly
no outgoing calls
no test coverage detected