Fix GPU test (#190)

By fixing subviewImpl.
Also, reeanble parfor test

Author: Hardcode84

numba_dpcomp/numba_dpcomp/mlir/tests/test_numba_parfor.py

@@ -78,7 +78,6 @@ def _gen_tests():
         "test_prange27",  # Literal return issue
         "test_simple01",  # Empty shape not failed
         "test_kmeans",  # List suport
-        "test_simple14",  # Slice shape mismatch
         "test_ndarray_fill",  # array.fill
         "test_fuse_argmin_argmax_max_min",  # numpy argmin, argmax
         "test_max",  # max reduction

numba_dpcomp/numba_dpcomp/mlir_compiler/lib/py_linalg_resolver.cpp

@@ -1439,7 +1439,17 @@ py::object subviewImpl(py::capsule context, py::handle src, py::handle offsets,
       for (auto i : llvm::seq(size_t(0), ret.size())) {
         auto dim = builder.createOrFold<mlir::tensor::DimOp>(
             loc, origSrcVal, static_cast<int64_t>(i));
-        auto offset = offsetVals[i].get<mlir::Value>();
+        auto offset = [&]() -> mlir::Value {
+          auto off = offsetVals[i];
+          if (off.is<mlir::Value>())
+            return off.get<mlir::Value>();
+
+          auto val = off.get<mlir::Attribute>()
+                         .cast<mlir::IntegerAttr>()
+                         .getValue()
+                         .getSExtValue();
+          return builder.create<mlir::arith::ConstantIndexOp>(loc, val);
+        }();
         auto size = builder.createOrFold<mlir::arith::SubIOp>(loc, dim, offset);
         ret[i] = size;
       }