diff --git a/mlir/lib/Target/LLVM/NVVM/Target.cpp b/mlir/lib/Target/LLVM/NVVM/Target.cpp index e75547ff9b850..acb903aa37caf 100644 --- a/mlir/lib/Target/LLVM/NVVM/Target.cpp +++ b/mlir/lib/Target/LLVM/NVVM/Target.cpp @@ -267,9 +267,12 @@ NVPTXSerializer::compileToBinary(const std::string &ptxCode) { std::optional ptxasCompiler = findTool("ptxas"); if (!ptxasCompiler) return std::nullopt; - std::optional fatbinaryTool = findTool("fatbinary"); - if (createFatbin && !fatbinaryTool) - return std::nullopt; + std::optional fatbinaryTool; + if (createFatbin) { + fatbinaryTool = findTool("fatbinary"); + if (!fatbinaryTool) + return std::nullopt; + } Location loc = getOperation().getLoc(); // Base name for all temp files: mlir---.