[SYCL] Add assume in size_t conversion operator

Sergey Kanaev · Sergey Kanaev · commit e334ad817aea · 2020-05-15T14:56:03.000+03:00
Signed-off-by: Sergey Kanaev &lt;sergey.kanaev@intel.com&gt;
diff --git a/sycl/include/CL/sycl/id.hpp b/sycl/include/CL/sycl/id.hpp
@@ -97,7 +97,9 @@ template <int dimensions = 1> class id : public detail::array<dimensions> {
    * int a = id<1>(value); */
 
   operator EnableIfT<(dimensions == 1), size_t>() const {
-    return this->common_array[0];
+    size_t Result = this->common_array[0];
+    __SYCL_ASSUME_INT(Result);
+    return Result;
   }
 #endif // __SYCL_DISABLE_ID_TO_INT_CONV__
 
diff --git a/sycl/test/basic_tests/id_queries_fit_int.cpp b/sycl/test/basic_tests/id_queries_fit_int.cpp
@@ -39,6 +39,13 @@ int main() {
   // CHECK: call void @llvm.assume(i1 {{.*}})
   int LocalRange = TestNDItem.get_local_range(0);
 
+  int GlobalIdConverted = TestNDItem.get_global_id();
+  // CHECK: call void @llvm.assume(i1 {{.*}})
+  int LocalIdConverted = TestNDItem.get_local_id();
+  // CHECK: call void @llvm.assume(i1 {{.*}})
+  int OffsetConferted = TestNDItem.get_offset();
+  // CHECK: call void @llvm.assume(i1 {{.*}})
+
   return 0;
 }
 // CHECK: }

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,9 @@ template <int dimensions = 1> class id : public detail::array<dimensions> {`
`97`	`97`	`* int a = id<1>(value); */`
`98`	`98`
`99`	`99`	`operator EnableIfT<(dimensions == 1), size_t>() const {`
`100`		`- return this->common_array[0];`
	`100`	`+ size_t Result = this->common_array[0];`
	`101`	`+ __SYCL_ASSUME_INT(Result);`
	`102`	`+ return Result;`
`101`	`103`	`}`
`102`	`104`	`#endif // __SYCL_DISABLE_ID_TO_INT_CONV__`
`103`	`105`