()
| 57 | |
| 58 | |
| 59 | def test_initial_A_single_cta(): |
| 60 | A = initial_A(warp_ext=16) |
| 61 | assert A.laneid == AxisRange(32, 0) |
| 62 | assert A.warpid == AxisRange(16, 0) |
| 63 | assert A.cta_id == AxisRange(1, 0) |
| 64 | assert A.size == 512 |
| 65 | |
| 66 | |
| 67 | def test_initial_A_cluster(): |