diff --git a/clang/test/dpct/thrust_sort_by_key_usm.cu b/clang/test/dpct/thrust_sort_by_key_usm.cu
index ae2d35a3ad7f..cad35176156c 100644
--- a/clang/test/dpct/thrust_sort_by_key_usm.cu
+++ b/clang/test/dpct/thrust_sort_by_key_usm.cu
@@ -43,16 +43,6 @@ int main(void) {
   thrust::sort_by_key(                h_ptr,      h_ptr+4,  BH.begin());
   thrust::sort_by_key(                d_ptr,      d_ptr+4,  BD.begin());
 
-//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-// CHECK:dpct::sort(oneapi::dpl::execution::seq, AH.begin(), AH.end(), BH.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::sort(oneapi::dpl::execution::make_device_policy(q_ct1), AD.begin(), AD.end(), BD.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::sort(oneapi::dpl::execution::seq, h_ptr, h_ptr + 4, BH.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::sort(oneapi::dpl::execution::seq, d_ptr, d_ptr + 4, BD.begin(), std::greater<int>());
-  // VERSION                          first       last      result      comparator
-  thrust::sort_by_key(                AH.begin(), AH.end(), BH.begin(), thrust::greater<int>());
-  thrust::sort_by_key(                AD.begin(), AD.end(), BD.begin(), thrust::greater<int>());
-  thrust::sort_by_key(                h_ptr,      h_ptr+4,  BH.begin(), thrust::greater<int>());  
-  thrust::sort_by_key(                d_ptr,      d_ptr+4,  BD.begin(), thrust::greater<int>());
 
 //////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 // CHECK:dpct::sort(oneapi::dpl::execution::seq, AH.begin(), AH.end(), BH.begin());
@@ -65,16 +55,6 @@ int main(void) {
   thrust::sort_by_key(thrust::host,   h_ptr,      h_ptr+4,  BH.begin());
   thrust::sort_by_key(thrust::device, d_ptr,      d_ptr+4,  BD.begin());
 
-//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-// CHECK:dpct::sort(oneapi::dpl::execution::seq, AH.begin(), AH.end(), BH.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::sort(oneapi::dpl::execution::make_device_policy(q_ct1), AD.begin(), AD.end(), BD.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::sort(oneapi::dpl::execution::seq, h_ptr, h_ptr + 4, BH.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::sort(oneapi::dpl::execution::make_device_policy(q_ct1), d_ptr, d_ptr + 4, BD.begin(), std::greater<int>());
-  // VERSION          exec            first       last      result      comparator
-  thrust::sort_by_key(thrust::host,   AH.begin(), AH.end(), BH.begin(), thrust::greater<int>());
-  thrust::sort_by_key(thrust::device, AD.begin(), AD.end(), BD.begin(), thrust::greater<int>());
-  thrust::sort_by_key(thrust::host,   h_ptr,      h_ptr+4,  BH.begin(), thrust::greater<int>());
-  thrust::sort_by_key(thrust::device, d_ptr,      d_ptr+4,  BD.begin(), thrust::greater<int>());
 
   return 0;
 }
diff --git a/clang/test/dpct/thrust_stable_sort_by_key_usm.cu b/clang/test/dpct/thrust_stable_sort_by_key_usm.cu
index be7499dae83c..c52b26dd540e 100644
--- a/clang/test/dpct/thrust_stable_sort_by_key_usm.cu
+++ b/clang/test/dpct/thrust_stable_sort_by_key_usm.cu
@@ -36,41 +36,23 @@ int main(void) {
 // CHECK:dpct::stable_sort(oneapi::dpl::execution::seq, AH.begin(), AH.end(), BH.begin());
 // CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::make_device_policy(q_ct1), AD.begin(), AD.end(), BD.begin());
 // CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::seq, h_ptr, h_ptr + 4, BH.begin());
+// CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::seq, d_ptr, d_ptr + 4, BD.begin());
   // VERSION                                 first       last      result
   thrust::stable_sort_by_key(                AH.begin(), AH.end(), BH.begin());
   thrust::stable_sort_by_key(                AD.begin(), AD.end(), BD.begin());
   thrust::stable_sort_by_key(                h_ptr,      h_ptr+4,  BH.begin());
+  thrust::stable_sort_by_key(                d_ptr,      d_ptr+4,  BD.begin());
 
 //////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 // CHECK:dpct::stable_sort(oneapi::dpl::execution::seq, AH.begin(), AH.end(), BH.begin(), std::greater<int>());
 // CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::make_device_policy(q_ct1), AD.begin(), AD.end(), BD.begin(), std::greater<int>());
 // CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::seq, h_ptr, h_ptr + 4, BH.begin(), std::greater<int>());
+// CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::seq, d_ptr, d_ptr + 4, BD.begin(), std::greater<int>());
   // VERSION                                 first       last      result      comparator
   thrust::stable_sort_by_key(                AH.begin(), AH.end(), BH.begin(), thrust::greater<int>());
   thrust::stable_sort_by_key(                AD.begin(), AD.end(), BD.begin(), thrust::greater<int>());
   thrust::stable_sort_by_key(                h_ptr,      h_ptr+4,  BH.begin(), thrust::greater<int>());  
-
-//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-// CHECK:dpct::stable_sort(oneapi::dpl::execution::seq, AH.begin(), AH.end(), BH.begin());
-// CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::make_device_policy(q_ct1), AD.begin(), AD.end(), BD.begin());
-// CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::seq, h_ptr, h_ptr + 4, BH.begin());
-// CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::make_device_policy(q_ct1), d_ptr, d_ptr + 4, BD.begin());
-  // VERSION                 exec            first       last      result
-  thrust::stable_sort_by_key(thrust::host,   AH.begin(), AH.end(), BH.begin());
-  thrust::stable_sort_by_key(thrust::device, AD.begin(), AD.end(), BD.begin());
-  thrust::stable_sort_by_key(thrust::host,   h_ptr,      h_ptr+4,  BH.begin());
-  thrust::stable_sort_by_key(thrust::device, d_ptr,      d_ptr+4,  BD.begin());
-
-//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-// CHECK:dpct::stable_sort(oneapi::dpl::execution::seq, AH.begin(), AH.end(), BH.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::make_device_policy(q_ct1), AD.begin(), AD.end(), BD.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::seq, h_ptr, h_ptr + 4, BH.begin(), std::greater<int>());
-// CHECK-NEXT:dpct::stable_sort(oneapi::dpl::execution::make_device_policy(q_ct1), d_ptr, d_ptr + 4, BD.begin(), std::greater<int>());
-  // VERSION                 exec            first       last      result      comparator
-  thrust::stable_sort_by_key(thrust::host,   AH.begin(), AH.end(), BH.begin(), thrust::greater<int>());
-  thrust::stable_sort_by_key(thrust::device, AD.begin(), AD.end(), BD.begin(), thrust::greater<int>());
-  thrust::stable_sort_by_key(thrust::host,   h_ptr,      h_ptr+4,  BH.begin(), thrust::greater<int>());
-  thrust::stable_sort_by_key(thrust::device, d_ptr,      d_ptr+4,  BD.begin(), thrust::greater<int>());
+  thrust::stable_sort_by_key(                d_ptr,      d_ptr+4,  BD.begin(), thrust::greater<int>());
 
   return 0;
 }