()
| 120 | |
| 121 | @T.prim_func(check_well_formed=False) |
| 122 | def test5(): |
| 123 | T.device_entry() |
| 124 | bx, by, bz = T.cta_id([8, 10, 12]) |
| 125 | cbx, cby, cbz = T.cta_id_in_cluster([2, 2, 1]) |
| 126 | clx, cly, clz = T.cluster_id([3, 5, 12]) |
| 127 | T.evaluate(bx + by + bz) |
| 128 | T.evaluate(cbx + cby + cbz) |
| 129 | T.evaluate(clx + cly + clz) |
| 130 | |
| 131 | @T.prim_func(check_well_formed=False) |
| 132 | def test6(): |