(A_ptr: T.handle)
| 40 | # fmt: off |
| 41 | @T.prim_func |
| 42 | def func(A_ptr: T.handle): |
| 43 | A = T.match_buffer(A_ptr, (10,), "int32") |
| 44 | |
| 45 | T.device_entry() |
| 46 | cta_id = T.cta_id([1]) |
| 47 | tid = T.thread_id([32]) |
| 48 | for i in T.serial(10): |
| 49 | if i == 2: |
| 50 | continue |
| 51 | if i == 7: |
| 52 | break |
| 53 | A[i] = i |
| 54 | # fmt: on |
| 55 | |
| 56 | expected = np.array([0, 1, 0, 3, 4, 5, 6, 0, 0, 0], dtype="int32") |
| 57 | run_test_break_continue(func, (10,), expected) |
nothing calls this directly
no test coverage detected
searching dependent graphs…