[SYCL] Allow gcc asm statements in kernel code (#1366)

Signed-off-by: Premanand M Rao <[email protected]>

Author: premanandrao

clang/lib/Sema/Sema.cpp

@@ -1639,9 +1639,10 @@ Sema::DeviceDiagBuilder Sema::targetDiag(SourceLocation Loc, unsigned DiagID) {
   if (getLangOpts().CUDA)
     return getLangOpts().CUDAIsDevice ? CUDADiagIfDeviceCode(Loc, DiagID)
                                       : CUDADiagIfHostCode(Loc, DiagID);
-  // TODO: analyze which usages of targetDiag could be reused for SYCL.
-  // if (getLangOpts().SYCLIsDevice)
-  //   return SYCLDiagIfDeviceCode(Loc, DiagID);
+
+  if (getLangOpts().SYCLIsDevice)
+    return SYCLDiagIfDeviceCode(Loc, DiagID);
+
   return DeviceDiagBuilder(DeviceDiagBuilder::K_Immediate, Loc, DiagID,
                            getCurFunctionDecl(), *this);
 }

clang/test/SemaSYCL/inline-asm.cpp

@@ -1,7 +1,35 @@
 // RUN: %clang_cc1 -fsycl -fsycl-is-device -fsyntax-only -verify %s -DLINUX_ASM
+// RUN: %clang_cc1 -fsycl -fsycl-is-device -fsyntax-only -verify %s -DLINUX_ASM -DSPIR_CHECK -triple spir64-unknown-unknown-sycldevice
 // RUN: %clang_cc1 -fsycl -fsycl-is-device -fsyntax-only -verify -triple x86_64-windows -fasm-blocks %s
 
-// expected-no-diagnostics
+#ifndef SPIR_CHECK
+//expected-no-diagnostics
+#endif // SPIR_CHECK
+
+static __inline unsigned int
+asm_func(unsigned int __leaf, unsigned long __d[]) {
+  unsigned int __result;
+#ifdef SPIR_CHECK
+  __asm__("enclu"
+          : "=a"(__result), "=b"(__d[0]), "=c"(__d[1]), "=d"(__d[2])
+          : "a"(__leaf), "b"(__d[0]), "c"(__d[1]), "d"(__d[2])
+          : "cc");
+#endif // SPIR_CHECK
+  return __result;
+}
+
+static __inline unsigned int
+asm_func_2(unsigned int __leaf, unsigned long __d[]) {
+  unsigned int __result;
+#ifdef SPIR_CHECK
+  //expected-error@+2 {{invalid output constraint '=a' in asm}}
+  __asm__("enclu"
+          : "=a"(__result), "=b"(__d[0]), "=c"(__d[1]), "=d"(__d[2])
+          : "a"(__leaf), "b"(__d[0]), "c"(__d[1]), "d"(__d[2])
+          : "cc");
+#endif // SPIR_CHECK
+  return __result;
+}
 
 void foo() {
   int a;
@@ -26,6 +54,13 @@ __attribute__((sycl_kernel)) void kernel_single_task(Func kernelFunc) {
   kernelFunc();
 #ifdef LINUX_ASM
   __asm__("int3");
+
+#ifdef SPIR_CHECK
+  unsigned int i = 3;
+  unsigned long d[4];
+  //expected-note@+1 {{called by 'kernel_single_task'}}
+  asm_func_2(i, d);
+#endif // SPIR_CHECK
 #else
   __asm int 3
 #endif // LINUX_ASM