diff --git a/src/CUDAStream.h b/src/CUDAStream.h index 14f2cc2..451cfc1 100644 --- a/src/CUDAStream.h +++ b/src/CUDAStream.h @@ -6,7 +6,7 @@ template class CUDAStream : public Stream { - private: + protected: // Size of arrays unsigned int array_size; // Device side pointers to arrays @@ -19,10 +19,10 @@ class CUDAStream : public Stream CUDAStream(const unsigned int); - void copy(); - void add(); - void mul(); - void triad(); + virtual void copy() override; + virtual void add() override; + virtual void mul() override; + virtual void triad() override; void write_arrays(const std::vector& a, const std::vector& b, const std::vector& c); void read_arrays(std::vector& a, std::vector& b, std::vector& c);