()
| 18 | |
| 19 | |
| 20 | def test_solver(): |
| 21 | from modules import devices, shared |
| 22 | try: |
| 23 | a = torch.randn(50, 50).to(device=devices.device, dtype=torch.float32) |
| 24 | b = torch.randn(50, 2).to(device=devices.device, dtype=torch.float32) |
| 25 | _x = torch.linalg.solve(a, b) |
| 26 | return True |
| 27 | except Exception as e: |
| 28 | log.debug(f'FramePack: solver=cpu {e}') |
| 29 | return False |
| 30 | |
| 31 | |
| 32 | def linalg_solve(A, B, device): |
no test coverage detected