(expect, fn, a, b, label, enabled=True)
| 769 | |
| 770 | |
| 771 | def run_test(expect, fn, a, b, label, enabled=True): |
| 772 | print(f" {label}: ...", end="") |
| 773 | if enabled: |
| 774 | actual = fn(a, b) |
| 775 | passed = torch.allclose(expect, actual.to(expect.dtype), atol=1.0) |
| 776 | icon = "✅" if passed else "❌" |
| 777 | else: |
| 778 | icon = "⭕" |
| 779 | print(f"\r {label}: {icon} ") |
| 780 | |
| 781 | |
| 782 | def validate(M, N, K, dtype): |