updating imex (#77)

* updating imex and CI

Author: fschlimb

.github/workflows/ci.yml

@@ -6,7 +6,7 @@ defaults:
     shell: bash
 jobs:
   build_and_test:
-    runs-on: [self-hosted, dds-base]
+    runs-on: tpi-ubuntu-latest
     strategy:
       fail-fast: false
     steps:
@@ -30,13 +30,14 @@ jobs:
         id: cache-miniconda
         uses: actions/cache@v3
         env:
-          MINICONDA_CACHE_NUMBER: 1  # Increase to reset cache
+          MINICONDA_CACHE_NUMBER: 2  # Increase to reset cache
         with:
           path: third_party/install/miniconda/**
           key: ${{ runner.os }}-miniconda-${{ env.MINICONDA_CACHE_NUMBER }}-${{ hashFiles('conda-env.yml') }}
       - name: Miniconda
         if: steps.cache-miniconda.outputs.cache-hit != 'true'
         run: |
+          rm -rf "$GITHUB_WORKSPACE"/third_party/install/miniconda
           mkdir -p "$GITHUB_WORKSPACE"/third_party/install
           cd "$GITHUB_WORKSPACE"/third_party
           wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

imex_version.txt

@@ -1 +1 @@
-a84e52b9c5074a71f9b935cf7f0c0384b462f3bd
+2875ae696359f267571a77ea774d81659a10d667

src/ManipOp.cpp

@@ -153,7 +153,7 @@ struct DeferredToDevice : public Deferred {
 
   bool generate_mlir(::mlir::OpBuilder &builder, const ::mlir::Location &loc,
                      jit::DepManager &dm) override {
-    auto av = dm.getDependent(builder, _a);
+    auto av = dm.getDependent(builder, Registry::get(_a));
 
     auto srcType = av.getType().dyn_cast<::imex::ndarray::NDArrayType>();
     assert(srcType);
@@ -183,12 +183,12 @@ struct DeferredToDevice : public Deferred {
                      void *r_allocated, void *r_aligned, intptr_t r_offset,
                      const intptr_t *r_sizes, const intptr_t *r_strides,
                      uint64_t *lo_allocated, uint64_t *lo_aligned) {
-                auto t = mk_tnsr(reinterpret_cast<Transceiver *>(this->team()),
-                                 _dtype, this->shape(), l_allocated, l_aligned,
-                                 l_offset, l_sizes, l_strides, o_allocated,
-                                 o_aligned, o_offset, o_sizes, o_strides,
-                                 r_allocated, r_aligned, r_offset, r_sizes,
-                                 r_strides, lo_allocated, lo_aligned);
+                auto t = mk_tnsr(this->guid(), this->dtype(), this->shape(),
+                                 this->device(), this->team(), l_allocated,
+                                 l_aligned, l_offset, l_sizes, l_strides,
+                                 o_allocated, o_aligned, o_offset, o_sizes,
+                                 o_strides, r_allocated, r_aligned, r_offset,
+                                 r_sizes, r_strides, lo_allocated, lo_aligned);
                 this->set_value(std::move(t));
               });
     return false;