Lines Matching refs:step

7 // CHECK-SAME: (%[[lb:.*]]: index, %[[ub:.*]]: index, %[[step:.*]]: index)
8 func.func @map1d(%lb: index, %ub: index, %step: index) {
12   // CHECK: %[[thread_offset:.+]] = affine.apply #[[mul_map]]()[%[[threads]]#0, %[[step]]]
14   // CHECK: %[[new_step:.+]] = affine.apply #[[mul_map]]()[%[[threads]]#1, %[[step]]]
16   // CHECK: scf.for %{{.*}} = %[[new_lb]] to %[[ub]] step %[[new_step]] {
17   scf.for %i = %lb to %ub step %step {}
22 // CHECK-SAME: (%[[lb:.*]]: index, %[[ub:.*]]: index, %[[step:.*]]: index)
23 func.func @map2d(%lb : index, %ub : index, %step : index) {
36   // thread_offset = step * (threadIdx.x + blockIdx.x * blockDim.x)
37   // CHECK: %[[thread_offset:.+]] = affine.apply #[[mul_map]]()[%[[tidxpbidxXbdimx]], %[[step]]]
42   // stepXgdimx = step * gridDim.x
43   // CHECK: %[[stepXgdimx:.+]] = affine.apply #[[mul_map]]()[%[[blocks]]#1, %[[step]]]
45   // new_step = step * gridDim.x * blockDim.x
48   // CHECK: scf.for %{{.*}} = %[[new_lb]] to %[[ub]] step %[[new_step]] {
50   scf.for %i = %lb to %ub step %step {}