diff options
| -rw-r--r-- | tgsi/lib/SOURCES | 3 | ||||
| -rw-r--r-- | tgsi/lib/workitem/get_group_id.ll | 18 | ||||
| -rw-r--r-- | tgsi/lib/workitem/get_local_size.ll | 18 | ||||
| -rw-r--r-- | tgsi/lib/workitem/get_num_groups.ll | 18 |
4 files changed, 57 insertions, 0 deletions
diff --git a/tgsi/lib/SOURCES b/tgsi/lib/SOURCES index c116b86..4aba61d 100644 --- a/tgsi/lib/SOURCES +++ b/tgsi/lib/SOURCES @@ -1 +1,4 @@ +workitem/get_group_id.ll workitem/get_local_id.ll +workitem/get_local_size.ll +workitem/get_num_groups.ll diff --git a/tgsi/lib/workitem/get_group_id.ll b/tgsi/lib/workitem/get_group_id.ll new file mode 100644 index 0000000..f0809af --- /dev/null +++ b/tgsi/lib/workitem/get_group_id.ll @@ -0,0 +1,18 @@ +declare i32 @llvm.tgsi.read.blockid.x() nounwind readnone +declare i32 @llvm.tgsi.read.blockid.y() nounwind readnone +declare i32 @llvm.tgsi.read.blockid.z() nounwind readnone + +define i32 @get_group_id(i32 %dim) nounwind readnone alwaysinline { + switch i32 %dim, label %default [i32 0, label %x_dim i32 1, label %y_dim i32 2, label %z_dim] +x_dim: + %x = call i32 @llvm.tgsi.read.blockid.x() nounwind readnone + ret i32 %x +y_dim: + %y = call i32 @llvm.tgsi.read.blockid.y() nounwind readnone + ret i32 %y +z_dim: + %z = call i32 @llvm.tgsi.read.blockid.z() nounwind readnone + ret i32 %z +default: + ret i32 0 +} diff --git a/tgsi/lib/workitem/get_local_size.ll b/tgsi/lib/workitem/get_local_size.ll new file mode 100644 index 0000000..8651dc9 --- /dev/null +++ b/tgsi/lib/workitem/get_local_size.ll @@ -0,0 +1,18 @@ +declare i32 @llvm.tgsi.read.blocksize.x() nounwind readnone +declare i32 @llvm.tgsi.read.blocksize.y() nounwind readnone +declare i32 @llvm.tgsi.read.blocksize.z() nounwind readnone + +define i32 @get_local_size(i32 %dim) nounwind readnone alwaysinline { + switch i32 %dim, label %default [i32 0, label %x_dim i32 1, label %y_dim i32 2, label %z_dim] +x_dim: + %x = call i32 @llvm.tgsi.read.blocksize.x() nounwind readnone + ret i32 %x +y_dim: + %y = call i32 @llvm.tgsi.read.blocksize.y() nounwind readnone + ret i32 %y +z_dim: + %z = call i32 @llvm.tgsi.read.blocksize.z() nounwind readnone + ret i32 %z +default: + ret i32 1 +} diff --git a/tgsi/lib/workitem/get_num_groups.ll b/tgsi/lib/workitem/get_num_groups.ll new file mode 100644 index 0000000..45cac88 --- /dev/null +++ b/tgsi/lib/workitem/get_num_groups.ll @@ -0,0 +1,18 @@ +declare i32 @llvm.tgsi.read.gridsize.x() nounwind readnone +declare i32 @llvm.tgsi.read.gridsize.y() nounwind readnone +declare i32 @llvm.tgsi.read.gridsize.z() nounwind readnone + +define i32 @get_num_groups(i32 %dim) nounwind readnone alwaysinline { + switch i32 %dim, label %default [i32 0, label %x_dim i32 1, label %y_dim i32 2, label %z_dim] +x_dim: + %x = call i32 @llvm.tgsi.read.gridsize.x() nounwind readnone + ret i32 %x +y_dim: + %y = call i32 @llvm.tgsi.read.gridsize.y() nounwind readnone + ret i32 %y +z_dim: + %z = call i32 @llvm.tgsi.read.gridsize.z() nounwind readnone + ret i32 %z +default: + ret i32 1 +} |