r600: Add image writing builtins.

Patch by: Zoltan Gilian

git-svn-id: https://llvm.org/svn/llvm-project/libclc/trunk@248161 91177308-0d34-0410-b5e6-96231b3b80d8

6 files changed, 90 insertions, 0 deletions

diff --git a/generic/include/clc/image/image.h b/generic/include/clc/image/image.h
index 8c9a444..0a37074 100644
--- a/generic/include/clc/image/image.h
+++ b/generic/include/clc/image/image.h
@@ -15,6 +15,13 @@ _CLC_OVERLOAD _CLC_DECL int get_image_channel_order (image3d_t image);
 _CLC_OVERLOAD _CLC_DECL int2 get_image_dim (image2d_t image);
 _CLC_OVERLOAD _CLC_DECL int4 get_image_dim (image3d_t image);
 
+_CLC_OVERLOAD _CLC_DECL void
+write_imagef(image2d_t image, int2 coord, float4 color);
+_CLC_OVERLOAD _CLC_DECL void
+write_imagei(image2d_t image, int2 coord, int4 color);
+_CLC_OVERLOAD _CLC_DECL void
+write_imageui(image2d_t image, int2 coord, uint4 color);
+
 _CLC_OVERLOAD _CLC_DECL float4
 read_imagef(image2d_t image, sampler_t sampler, int2 coord);
 _CLC_OVERLOAD _CLC_DECL float4
diff --git a/r600/lib/SOURCES b/r600/lib/SOURCES
index 706c879..029b22c 100644
--- a/r600/lib/SOURCES
+++ b/r600/lib/SOURCES
@@ -20,3 +20,7 @@ image/read_imagef.cl
 image/read_imagei.cl
 image/read_imageui.cl
 image/read_image_impl.ll
+image/write_imagef.cl
+image/write_imagei.cl
+image/write_imageui.cl
+image/write_image_impl.ll
diff --git a/r600/lib/image/write_image_impl.ll b/r600/lib/image/write_image_impl.ll
new file mode 100644
index 0000000..265f5d6
--- /dev/null
+++ b/r600/lib/image/write_image_impl.ll
@@ -0,0 +1,52 @@
+%opencl.image2d_t = type opaque
+%opencl.image3d_t = type opaque
+
+declare i32 @llvm.OpenCL.image.get.resource.id.2d(
+  %opencl.image2d_t addrspace(1)*) nounwind readnone
+declare i32 @llvm.OpenCL.image.get.resource.id.3d(
+  %opencl.image3d_t addrspace(1)*) nounwind readnone
+
+declare void @llvm.r600.rat.store.typed(<4 x i32> %color, <4 x i32> %coord, i32 %rat_id)
+
+define void @__clc_write_imageui_2d(
+    %opencl.image2d_t addrspace(1)* nocapture %img,
+    <2 x i32> %coord, <4 x i32> %color) #0 {
+
+  ; Coordinate int2 -> int4.
+  %e0 = extractelement <2 x i32> %coord, i32 0
+  %e1 = extractelement <2 x i32> %coord, i32 1
+  %coord.0 = insertelement <4 x i32> undef,    i32 %e0, i32 0
+  %coord.1 = insertelement <4 x i32> %coord.0, i32 %e1, i32 1
+  %coord.2 = insertelement <4 x i32> %coord.1, i32 0,  i32 2
+  %coord.3 = insertelement <4 x i32> %coord.2, i32 0,  i32 3
+
+  ; Get RAT ID.
+  %img_id = call i32 @llvm.OpenCL.image.get.resource.id.2d(
+      %opencl.image2d_t addrspace(1)* %img)
+  %rat_id = add i32 %img_id, 1
+
+  ; Call store intrinsic.
+  call void @llvm.r600.rat.store.typed(<4 x i32> %color, <4 x i32> %coord.3, i32 %rat_id)
+  ret void
+}
+
+define void @__clc_write_imagei_2d(
+    %opencl.image2d_t addrspace(1)* nocapture %img,
+    <2 x i32> %coord, <4 x i32> %color) #0 {
+  call void @__clc_write_imageui_2d(
+      %opencl.image2d_t addrspace(1)* nocapture %img,
+      <2 x i32> %coord, <4 x i32> %color)
+  ret void
+}
+
+define void @__clc_write_imagef_2d(
+    %opencl.image2d_t addrspace(1)* nocapture %img,
+    <2 x i32> %coord, <4 x float> %color) #0 {
+  %color.i32 = bitcast <4 x float> %color to <4 x i32>
+  call void @__clc_write_imageui_2d(
+      %opencl.image2d_t addrspace(1)* nocapture %img,
+      <2 x i32> %coord, <4 x i32> %color.i32)
+  ret void
+}
+
+attributes #0 = { alwaysinline }
diff --git a/r600/lib/image/write_imagef.cl b/r600/lib/image/write_imagef.cl
new file mode 100644
index 0000000..4483fcf
--- /dev/null
+++ b/r600/lib/image/write_imagef.cl
@@ -0,0 +1,9 @@
+#include <clc/clc.h>
+
+_CLC_DECL void __clc_write_imagef_2d(image2d_t image, int2 coord, float4 color);
+
+_CLC_OVERLOAD _CLC_DEF void
+write_imagef(image2d_t image, int2 coord, float4 color)
+{
+  __clc_write_imagef_2d(image, coord, color);
+}
diff --git a/r600/lib/image/write_imagei.cl b/r600/lib/image/write_imagei.cl
new file mode 100644
index 0000000..394a223
--- /dev/null
+++ b/r600/lib/image/write_imagei.cl
@@ -0,0 +1,9 @@
+#include <clc/clc.h>
+
+_CLC_DECL void __clc_write_imagei_2d(image2d_t image, int2 coord, int4 color);
+
+_CLC_OVERLOAD _CLC_DEF void
+write_imagei(image2d_t image, int2 coord, int4 color)
+{
+  __clc_write_imagei_2d(image, coord, color);
+}
diff --git a/r600/lib/image/write_imageui.cl b/r600/lib/image/write_imageui.cl
new file mode 100644
index 0000000..91344de
--- /dev/null
+++ b/r600/lib/image/write_imageui.cl
@@ -0,0 +1,9 @@
+#include <clc/clc.h>
+
+_CLC_DECL void __clc_write_imageui_2d(image2d_t image, int2 coord, uint4 color);
+
+_CLC_OVERLOAD _CLC_DEF void
+write_imageui(image2d_t image, int2 coord, uint4 color)
+{
+  __clc_write_imageui_2d(image, coord, color);
+}