Method UnifiedVectorWrapper

ngscuda/unifiedvector.cpp:363–392

  }

  UnifiedVectorWrapper :: UnifiedVectorWrapper(const BaseVector & vec_, optional<IntRange> opt_range)
    : vec(vec_)
  {
    IntRange range = {0, vec.Size()};
    if(opt_range)
      range = *opt_range;
    this->size = range.Size();

    auto uptr = dynamic_cast<const UnifiedVector*>(&vec_);
    if(uptr)
    {
      host_data = uptr->HostData() + range.First();
      dev_data = uptr->DevData() + range.First();
      uptr->UpdateDevice();
      uptr->InvalidateHost();
      initial_host_uptodate =  uptr->IsHostUptodate();
      initial_dev_uptodate =  uptr->IsDevUptodate();
    }
    else
      {
      auto err = cudaMalloc((void**)&dev_data, size*sizeof(double));
      if (err != 0)
        throw Exception("UnifiedVector allocation error, ec="+ToString(err));
      initial_host_uptodate = true;
      initial_dev_uptodate = false;
      host_data = vec.FVDouble().Data() + range.First();
    }
    host_uptodate = initial_host_uptodate;
    dev_uptodate = initial_dev_uptodate;
  }

  UnifiedVectorWrapper :: ~UnifiedVectorWrapper()
  {