Fix cublas transpose

EnzymeAD · Oct 26, 2023 · 6eb2e19 · 6eb2e19
1 parent de562f2
commit 6eb2e19
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/enzyme/Enzyme/Utils.cpp b/enzyme/Enzyme/Utils.cpp
@@ -2637,10 +2637,11 @@ llvm::Value *transpose(llvm::IRBuilder<> &B, llvm::Value *V, bool byRef,
     }
 
     // cblas
-    return B.CreateSelect(
-        B.CreateICmpEQ(V, ConstantInt::get(V->getType(), 111)),
-        ConstantInt::get(V->getType(), 112),
-        ConstantInt::get(V->getType(), 111));
+    if (!cublas)
+      return B.CreateSelect(
+          B.CreateICmpEQ(V, ConstantInt::get(V->getType(), 111)),
+          ConstantInt::get(V->getType(), 112),
+          ConstantInt::get(V->getType(), 111));
   }
 
   if (byRef) {