Method Project

ngscuda/cuda_linalg.cpp:912–948

  }

  void DevProjector :: Project (BaseVector & x) const
  {
    static Timer t("DevProjector::Project"); RegionTimer reg(t);
    if (x.EntrySize() != 1)
      throw Exception("DevProjector :: Project not implemented for EntrySize > 1");

    UnifiedVectorWrapper ux(x);

    ux.UpdateDevice();

    // DevProjectorProject (bits->Size(), (double*)ux.DevData(), bits->Data(), keep_values);
    if (keep_values)
      DeviceParallelFor
        (bits->Size(),
         [x=ux.DevData(), bits=bits->Data()] DEVICE_LAMBDA (auto i)
         {
           unsigned char mask = (char(1) << (i % CHAR_BIT));
           unsigned int addr = i / CHAR_BIT;
           
           if (! (bits[addr] & mask) )
             x[i] = 0.0;
         });
    else
      DeviceParallelFor
        (bits->Size(),
         [x=ux.DevData(), bits=bits->Data()] DEVICE_LAMBDA (auto i)
         {
           unsigned char mask = (char(1) << (i % CHAR_BIT));
           unsigned int addr = i / CHAR_BIT;
           
           if (bits[addr] & mask)
             x[i] = 0;
         });
      
      
    ux.InvalidateHost();
  }