ignore cpu validation to WA GOMP_CPU_AFFINITY stuck in cpu validation code

example/cpu_01_conv2d_fwd/cpu_conv2d_fwd.cpp

@@ -415,6 +415,8 @@ int main(int argc, char* argv[])
         wei_device_buf.ToDevice(wei_y_x_c_k.mData.data());
 #endif
         // get host result
+        int cpu_validation = ck::getenv_int("CK_CPU_VALIDATION", 1);
+        if(cpu_validation)
         {
             auto ref_conv    = ReferenceConvFwdInstance{};
             auto ref_invoker = ref_conv.MakeInvoker();
@@ -610,7 +612,7 @@ int main(int argc, char* argv[])
 
                 out_device_buf.FromDevice(out_n_k_ho_wo_device_result.mData.data());
 
-                if(!check_out(out_n_k_ho_wo_host_result,
+                if(cpu_validation && !check_out(out_n_k_ho_wo_host_result,
                                                 out_n_k_ho_wo_device_result,
                                                 1e-6,
                                                 per_pixel_check))
@@ -620,7 +622,8 @@ int main(int argc, char* argv[])
                 }
                 else
                 {
-                    std::cout << "Pass Info: " << conv_ptr->GetTypeString() << ", Time:" << time
+                    std::cout << (cpu_validation ? "Pass" : "Ignore")
+                              << " Info: " << conv_ptr->GetTypeString() << ", Time:" << time
                               << "ms, Gflops:" << gflops << std::endl;
 
                     if(time < fastest_kernel_time)

example/cpu_02_conv2d_fwd_bias_relu_add/cpu_conv2d_fwd_bias_relu_add.cpp

@@ -452,6 +452,8 @@ int main(int argc, char* argv[])
         resi_device_buf.ToDevice(residual.mData.data());
 
         // get host result
+        int cpu_validation = ck::getenv_int("CK_CPU_VALIDATION", 1);
+        if(cpu_validation)
         {
             auto ref_conv    = ReferenceConvFwdInstance{};
             auto ref_invoker = ref_conv.MakeInvoker();
@@ -748,7 +750,7 @@ int main(int argc, char* argv[])
 
                 out_device_buf.FromDevice(out_n_k_ho_wo_device_result.mData.data());
 
-                if(!check_out(out_n_k_ho_wo_host_result,
+                if(cpu_validation && !check_out(out_n_k_ho_wo_host_result,
                                                 out_n_k_ho_wo_device_result,
                                                 1e-6,
                                                 per_pixel_check))
@@ -758,7 +760,8 @@ int main(int argc, char* argv[])
                 }
                 else
                 {
-                    std::cout << "Pass Info: " << conv_ptr->GetTypeString() << ", Time:" << time
+                    std::cout << (cpu_validation ? "Pass" : "Ignore")
+                              << " Info: " << conv_ptr->GetTypeString() << ", Time:" << time
                               << "ms, Gflops:" << gflops << std::endl;
 
                     if(time < fastest_kernel_time)