updating imex sha (#56)

* updating to new imex
* adding parallel runs of examples to CI

Author: fschlimb

.github/workflows/ci.yml

@@ -144,8 +144,14 @@ jobs:
           cd examples
           DDPT_FALLBACK=numpy python -u ./stencil-2d.py 5 1024 star 4
           DDPT_FALLBACK=numpy DDPT_FORCE_DIST=1 python -u ./stencil-2d.py 5 1024 star 4
-          python -u ./wave_equation.py -ct
-          DDPT_FORCE_DIST=1 python -u ./wave_equation.py -ct
+          mpirun -n 2 -genv DDPT_FALLBACK=numpy python -u ./stencil-2d.py 5 2048 star 2
+          mpirun -n 3 -genv DDPT_FALLBACK=numpy python -u ./stencil-2d.py 5 2048 star 2
+          mpirun -n 4 -genv DDPT_FALLBACK=numpy python -u ./stencil-2d.py 5 2048 star 2
+          python -u ./wave_equation.py
+          DDPT_FORCE_DIST=1 python -u ./wave_equation.py
+          mpirun -n 2 python -u ./wave_equation.py
+          mpirun -n 3 python -u ./wave_equation.py
+          mpirun -n 4 python -u ./wave_equation.py
           cd -
       - name: Cleanup
         run: |

conda-env.yml

@@ -41,6 +41,7 @@ dependencies:
   - libssh2=1.10.0=h8f2d780_0
   - libstdcxx-ng=11.2.0=h1234567_1
   - libuv=1.40.0=h7b6447c_2
+  - lit=14.0.6
   - lz4-c=1.9.4=h6a678d5_0
   - mkl=2023.1.0=intel_46342
   - mkl-service=2.4.0=py39h75d02e3_15

imex_version.txt

@@ -1 +1 @@
-e171d0b0a365bf2c6dde6f73be362b4cb997630f
+47768ada136b299effc408de420710fcacc12fda

src/Creator.cpp

@@ -43,11 +43,11 @@ struct DeferredFull : public Deferred {
       if (is_none(val)) {
         return {};
       } else if constexpr (std::is_floating_point_v<T>) {
-        return ::imex::createFloat<sizeof(T) * 8>(loc, builder, val._float);
+        return ::imex::createFloat(loc, builder, val._float, sizeof(T) * 8);
       } else if constexpr (std::is_same_v<bool, T>) {
-        return ::imex::createInt<1>(loc, builder, val._int);
+        return ::imex::createInt(loc, builder, val._int, 1);
       } else if constexpr (std::is_integral_v<T>) {
-        return ::imex::createInt<sizeof(T) * 8>(loc, builder, val._int);
+        return ::imex::createInt(loc, builder, val._int, sizeof(T) * 8);
       }
       assert("Unsupported dtype in dispatch");
       return {};

src/ManipOp.cpp

@@ -33,7 +33,7 @@ struct DeferredReshape : public Deferred {
     auto copyA =
         _copy == COPY_POSSIBLE
             ? ::mlir::IntegerAttr()
-            : ::imex::getIntAttr<1>(builder, COPY_ALWAYS ? true : false);
+            : ::imex::getIntAttr(builder, COPY_ALWAYS ? true : false, 1);
 
     auto outTyp = ::imex::ptensor::PTensorType::get(
         shape(), ::imex::dist::getElementType(av));