Fix typo and missing situation.

lattice · Oct 4, 2023 · 37a08ac · 37a08ac
1 parent 830eed0
commit 37a08ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/include/kernels/unitarize_links.cuh b/include/kernels/unitarize_links.cuh
@@ -236,7 +236,7 @@ namespace quda {
         if (result.isUnitary(arg.max_error) == false) atomic_fetch_add(arg.fails, 1);
       }
 
-      if (Arg::phase == QUDA_STAGGERED_PHASE_CHROMA) {  // Special unitrize the result for Chroma convention
+      if constexpr (Arg::phase == QUDA_STAGGERED_PHASE_CHROMA) {  // Special unitraize the result for Chroma convention
         int x[4];
         getCoords(x, x_cb, arg.X, parity);
 

diff --git a/lib/unitarize_links_quda.cu b/lib/unitarize_links_quda.cu
@@ -133,6 +133,9 @@ namespace quda {
       } else if (in.StaggeredPhase() == QUDA_STAGGERED_PHASE_CHROMA) {
         UnitarizeArg<Float, nColor, recon, QUDA_STAGGERED_PHASE_CHROMA> arg(out, in, fails, max_iter, unitarize_eps, max_error, reunit_allow_svd, reunit_svd_only, svd_rel_error, svd_abs_error);
         launch<Unitarize>(tp, stream, arg);
+      } else if (in.StaggeredPhase() == QUDA_STAGGERED_PHASE_NO) {
+        UnitarizeArg<Float, nColor, recon, QUDA_STAGGERED_PHASE_NO> arg(out, in, fails, max_iter, unitarize_eps, max_error, reunit_allow_svd, reunit_svd_only, svd_rel_error, svd_abs_error);
+        launch<Unitarize>(tp, stream, arg);
       } else {
         errorQuda("Undefined phase type %d", in.StaggeredPhase());
       }