Workgroup.cpp

Go to the documentation of this file.

// Copyright (c) 2018 the Talvos developers. All rights reserved.
//
// This file is distributed under a three-clause BSD license. For full license
// terms please see the LICENSE file distributed with this source code.


#include <spirv/unified1/spirv.h>

#include "PipelineExecutor.h"
#include "talvos/Invocation.h"
#include "talvos/Memory.h"
#include "talvos/Module.h"
#include "talvos/PipelineStage.h"
#include "talvos/Type.h"
#include "talvos/Variable.h"
#include "talvos/Workgroup.h"

namespace talvos
{

Workgroup::Workgroup(Device &Dev, const PipelineExecutor &Executor,
                     Dim3 GroupId)
{
  this->GroupId = GroupId;
  LocalMemory = new Memory(Dev, MemoryScope::Workgroup);

  const PipelineStage &Stage = Executor.getCurrentStage();

  // Allocate workgroup variables.
  for (auto V : Stage.getModule()->getVariables())
  {
    const Type *Ty = V->getType();
    if (Ty->getStorageClass() != SpvStorageClassWorkgroup)
      continue;

    size_t NumBytes = Ty->getElementType()->getSize();
    uint64_t Address = LocalMemory->allocate(NumBytes);
    Variables.push_back({V->getId(), Object(Ty, Address)});
  }
}

Workgroup::~Workgroup() { delete LocalMemory; }

void Workgroup::addWorkItem(std::unique_ptr<Invocation> WorkItem)
{
  WorkItems.push_back(std::move(WorkItem));
}

} // namespace talvos