Vertex.hpp

// Copyright (c) 2015 Graphcore Ltd. All rights reserved.
 
#ifndef poplar_Vertex_hpp_
#define poplar_Vertex_hpp_
 
#include <poplar/InOutTypes.hpp>
#include <poplar/TileConstants.hpp>
#include <poplar/VectorListTypes.hpp>
#include <poplar/VectorTypes.hpp>
 
#ifdef __IPU__
// Worker vertex state must be 4 byte aligned. There is currently no
// restriction on supervisor state alignment.
#define WORKER_ALIGN alignas(4)
#else
#define WORKER_ALIGN
#endif
 
namespace poplar {
 
class Vertex {};
 
class SupervisorVertex {};
 
class ExternalExchangeVertex {};
 
class MultiVertex {
public:
  static inline constexpr unsigned numWorkers() { return CTXT_WORKERS; }
};
 
template <bool pred>
using SupervisorVertexIf = std::conditional_t<pred, SupervisorVertex, Vertex>;
 
} // End namespace poplar.
 
#endif // poplar_Vertex_hpp_