triton/lib/driver/ndrange.cpp

#include <algorithm>
#include "isaac/driver/ndrange.h"

namespace isaac
{

namespace driver
{

NDRange::NDRange(size_t size0)
{
    sizes_[0] = size0;
    sizes_[1] = 1;
    sizes_[2] = 1;
    dimension_ = 1;
}

NDRange::NDRange(size_t size0, size_t size1)
{
    sizes_[0] = size0;
    sizes_[1] = size1;
    sizes_[2] = 1;
    dimension_ = 2;
}

NDRange::NDRange(size_t size0, size_t size1, size_t size2)
{
    sizes_[0] = size0;
    sizes_[1] = size1;
    sizes_[2] = size2;
    dimension_ = 3;
}

int NDRange::dimension() const
{
 return dimension_;
}

NDRange::operator const size_t*() const
{
  return (const size_t*) sizes_;
}

}

}