72 lines
1.9 KiB
C++
72 lines
1.9 KiB
C++
// Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
|
|
//
|
|
// For full license terms please see the LICENSE file distributed with this
|
|
// source code
|
|
|
|
#pragma once
|
|
|
|
#include <iostream>
|
|
#include <stdexcept>
|
|
#include "Stream.h"
|
|
|
|
#define IMPLEMENTATION_STRING "STD"
|
|
|
|
template <typename N = size_t>
|
|
class ranged {
|
|
N from, to;
|
|
public:
|
|
ranged(N from, N to ): from(from), to(to) {}
|
|
class iterator {
|
|
N num;
|
|
public:
|
|
using difference_type = N;
|
|
using value_type = N;
|
|
using pointer = const N*;
|
|
using reference = N&;
|
|
using iterator_category = std::random_access_iterator_tag;
|
|
explicit iterator(N _num = 0) : num(_num) {}
|
|
iterator& operator++() { num++; return *this; }
|
|
iterator operator++(int) { iterator retval = *this; ++(*this); return retval; }
|
|
bool operator==(iterator other) const { return num == other.num; }
|
|
bool operator!=(iterator other) const { return *this != other; }
|
|
reference operator*() const { return num;}
|
|
difference_type operator-(const iterator &it) const { return num - it.num; }
|
|
value_type operator[](const difference_type &i) const { return num+i; }
|
|
|
|
};
|
|
iterator begin() { return iterator(from); }
|
|
iterator end() { return iterator(to >= from? to+1 : to-1); }
|
|
};
|
|
|
|
template <class T>
|
|
class STDStream : public Stream<T>
|
|
{
|
|
protected:
|
|
// Size of arrays
|
|
int array_size;
|
|
|
|
// induction range
|
|
ranged<int> range;
|
|
|
|
// Device side pointers
|
|
std::vector<T> a;
|
|
std::vector<T> b;
|
|
std::vector<T> c;
|
|
|
|
|
|
public:
|
|
STDStream(const int, int) noexcept;
|
|
~STDStream() = default;
|
|
|
|
virtual void copy() override;
|
|
virtual void add() override;
|
|
virtual void mul() override;
|
|
virtual void triad() override;
|
|
virtual void nstream() override;
|
|
virtual T dot() override;
|
|
|
|
virtual void init_arrays(T initA, T initB, T initC) override;
|
|
virtual void read_arrays(std::vector<T>& a, std::vector<T>& b, std::vector<T>& c) override;
|
|
};
|
|
|