Example #1
0
        def udt(output):
            global_id = roc.get_global_id(0)
            global_size = roc.get_global_size(0)
            local_id = roc.get_local_id(0)
            group_id = roc.get_group_id(0)
            num_groups = roc.get_num_groups(0)
            workdim = roc.get_work_dim()
            local_size = roc.get_local_size(0)

            output[0, group_id, local_id] = global_id
            output[1, group_id, local_id] = global_size
            output[2, group_id, local_id] = local_id
            output[3, group_id, local_id] = local_size
            output[4, group_id, local_id] = group_id
            output[5, group_id, local_id] = num_groups
            output[6, group_id, local_id] = workdim
Example #2
0
 def udt2(output):
     g0 = roc.get_global_id(0)
     g1 = roc.get_global_id(1)
     output[g0, g1] = roc.get_work_dim()
Example #3
0
 def udt(output):
     global_id = roc.get_global_id(0)
     workdim = roc.get_work_dim()
     output[global_id] = workdim