Check optionals before getting KnownGrid/BlockSize

Author: AndreyPavlenko

lib/gc/Transforms/GPU/GpuToGpuOcl.cpp

@@ -403,15 +403,24 @@ struct ConvertLaunch final : ConvertOpPattern<gpu::LaunchFuncOp> {
     SmallVector<int32_t> argSize;
     kernelMod->walk([&](gpu::GPUFuncOp func) {
       if (func.getName() == gpuLaunch.getKernelName()) {
-        for (auto s : func.getKnownGridSize().value()) {
-          globalSize.emplace_back(s);
+        if (auto size = func.getKnownGridSize()) {
+          for (auto s : size.value()) {
+            globalSize.emplace_back(s);
+          }
         }
-        for (auto s : func.getKnownBlockSize().value()) {
-          localSize.emplace_back(s);
+        if (auto size = func.getKnownBlockSize()) {
+          for (auto s : size.value()) {
+            localSize.emplace_back(s);
+          }
         }
       }
     });
-    assert(globalSize.size() == 3 && localSize.size() == 3);
+    while (globalSize.size() < 3) {
+      globalSize.emplace_back(1);
+    }
+    while (localSize.size() < 3) {
+      localSize.emplace_back(1);
+    }
     globalSize = {globalSize[0] * localSize[0], globalSize[1] * localSize[1],
                   globalSize[2] * localSize[2]};
     for (auto arg : adaptor.getKernelOperands()) {