diff --git a/vtkm/rendering/MapperGL.cxx b/vtkm/rendering/MapperGL.cxx index f8f9a1a3f..0d223eabe 100644 --- a/vtkm/rendering/MapperGL.cxx +++ b/vtkm/rendering/MapperGL.cxx @@ -166,10 +166,11 @@ struct MapColorAndVerticesInvokeFunctor { VTKM_IS_DEVICE_ADAPTER_TAG(Device); - + vtkm::cont::Token token; MapColorAndVertices worklet( - this->ColorTable.PrepareForExecution(device), this->SMin, this->SDiff); - vtkm::worklet::DispatcherMapField dispatcher(worklet); + this->ColorTable.PrepareForExecution(device, token), this->SMin, this->SDiff); + vtkm::worklet::DispatcherMapField dispatcher(worklet); + dispatcher.SetDevice(Device{}); vtkm::cont::ArrayHandleIndex indexArray(this->TriangleIndices.GetNumberOfValues()); dispatcher.Invoke(indexArray,