[libclc] Add v3 variants of async_work_group_copy/async_work_group_strided_copy/prefetch (#137932)

wenju-he · web-flow · commit 5b6fc610919c · 2025-04-30T13:19:08.000+01:00
3-component vector type is supported for them per OpenCL spec.
diff --git a/libclc/generic/include/clc/async/async_work_group_copy.h b/libclc/generic/include/clc/async/async_work_group_copy.h
@@ -9,15 +9,21 @@
 #define __CLC_DST_ADDR_SPACE local
 #define __CLC_SRC_ADDR_SPACE global
 #define __CLC_BODY <clc/async/async_work_group_copy.inc>
-#include <clc/async/gentype.inc>
+#include <clc/integer/gentype.inc>
+#undef __CLC_BODY
+#define __CLC_BODY <clc/async/async_work_group_copy.inc>
+#include <clc/math/gentype.inc>
+#undef __CLC_BODY
 #undef __CLC_DST_ADDR_SPACE
 #undef __CLC_SRC_ADDR_SPACE
-#undef __CLC_BODY
 
 #define __CLC_DST_ADDR_SPACE global
 #define __CLC_SRC_ADDR_SPACE local
 #define __CLC_BODY <clc/async/async_work_group_copy.inc>
-#include <clc/async/gentype.inc>
+#include <clc/integer/gentype.inc>
+#undef __CLC_BODY
+#define __CLC_BODY <clc/async/async_work_group_copy.inc>
+#include <clc/math/gentype.inc>
+#undef __CLC_BODY
 #undef __CLC_DST_ADDR_SPACE
 #undef __CLC_SRC_ADDR_SPACE
-#undef __CLC_BODY
diff --git a/libclc/generic/include/clc/async/async_work_group_strided_copy.h b/libclc/generic/include/clc/async/async_work_group_strided_copy.h
@@ -9,15 +9,21 @@
 #define __CLC_DST_ADDR_SPACE local
 #define __CLC_SRC_ADDR_SPACE global
 #define __CLC_BODY <clc/async/async_work_group_strided_copy.inc>
-#include <clc/async/gentype.inc>
+#include <clc/integer/gentype.inc>
+#undef __CLC_BODY
+#define __CLC_BODY <clc/async/async_work_group_strided_copy.inc>
+#include <clc/math/gentype.inc>
+#undef __CLC_BODY
 #undef __CLC_DST_ADDR_SPACE
 #undef __CLC_SRC_ADDR_SPACE
-#undef __CLC_BODY
 
 #define __CLC_DST_ADDR_SPACE global
 #define __CLC_SRC_ADDR_SPACE local
 #define __CLC_BODY <clc/async/async_work_group_strided_copy.inc>
-#include <clc/async/gentype.inc>
+#include <clc/integer/gentype.inc>
+#undef __CLC_BODY
+#define __CLC_BODY <clc/async/async_work_group_strided_copy.inc>
+#include <clc/math/gentype.inc>
+#undef __CLC_BODY
 #undef __CLC_DST_ADDR_SPACE
 #undef __CLC_SRC_ADDR_SPACE
-#undef __CLC_BODY
diff --git a/libclc/generic/include/clc/async/gentype.inc b/libclc/generic/include/clc/async/gentype.inc
diff --git a/libclc/generic/include/clc/async/prefetch.h b/libclc/generic/include/clc/async/prefetch.h
@@ -7,5 +7,9 @@
 //===----------------------------------------------------------------------===//
 
 #define __CLC_BODY <clc/async/prefetch.inc>
-#include <clc/async/gentype.inc>
+#include <clc/integer/gentype.inc>
+#undef __CLC_BODY
+
+#define __CLC_BODY <clc/async/prefetch.inc>
+#include <clc/math/gentype.inc>
 #undef __CLC_BODY
diff --git a/libclc/generic/lib/async/async_work_group_copy.cl b/libclc/generic/lib/async/async_work_group_copy.cl
@@ -9,4 +9,9 @@
 #include <clc/clc.h>
 
 #define __CLC_BODY <async_work_group_copy.inc>
-#include <clc/async/gentype.inc>
+#include <clc/integer/gentype.inc>
+#undef __CLC_BODY
+
+#define __CLC_BODY <async_work_group_copy.inc>
+#include <clc/math/gentype.inc>
+#undef __CLC_BODY
diff --git a/libclc/generic/lib/async/async_work_group_strided_copy.cl b/libclc/generic/lib/async/async_work_group_strided_copy.cl
@@ -9,4 +9,9 @@
 #include <clc/clc.h>
 
 #define __CLC_BODY <async_work_group_strided_copy.inc>
-#include <clc/async/gentype.inc>
+#include <clc/integer/gentype.inc>
+#undef __CLC_BODY
+
+#define __CLC_BODY <async_work_group_strided_copy.inc>
+#include <clc/math/gentype.inc>
+#undef __CLC_BODY
diff --git a/libclc/generic/lib/async/prefetch.cl b/libclc/generic/lib/async/prefetch.cl
@@ -9,4 +9,9 @@
 #include <clc/clc.h>
 
 #define __CLC_BODY <prefetch.inc>
-#include <clc/async/gentype.inc>
+#include <clc/integer/gentype.inc>
+#undef __CLC_BODY
+
+#define __CLC_BODY <prefetch.inc>
+#include <clc/math/gentype.inc>
+#undef __CLC_BODY
