diff --git a/src/main.cpp b/src/main.cpp index 76c3da4..6702e11 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -1,18 +1,28 @@ #include +#include #include "common.h" #include "Stream.h" #include "CUDAStream.h" +const unsigned int ARRAY_SIZE = 52428800; + +#define IMPLEMENTATION_STRING "CUDA" int main(int argc, char *argv[]) { std::cout << "GPU-STREAM" << std::endl << "Version:" << VERSION_STRING << std::endl - << "Implementation: " << std::endl; + << "Implementation: " << IMPLEMENTATION_STRING << std::endl; + + + // Create host vectors + std::vector a(ARRAY_SIZE, 1.0); + std::vector b(ARRAY_SIZE, 2.0); + std::vector c(ARRAY_SIZE, 0.0); Stream *stream; stream = new CUDAStream();