(self)
| 40 | self.state = 0 |
| 41 | |
| 42 | def offload(self): |
| 43 | if self.state == 1 and ( |
| 44 | self.offload_dtype != self.onload_dtype |
| 45 | or self.offload_device != self.onload_device |
| 46 | ): |
| 47 | self.module.to(dtype=self.offload_dtype, device=self.offload_device) |
| 48 | self.state = 0 |
| 49 | |
| 50 | def onload(self): |
| 51 | if self.state == 0 and ( |
no outgoing calls
no test coverage detected