1e94a41f3c
PRK has a nstream kernel, which is Triad with a += update. This means there are 3 reads and a write, which is a higher read/write ratio. In addition, non-temporal stores for the write on CPUs will not be beneficial, and so compilers should take care to emit these for the other kernels, but not these.
47 lines
983 B
C++
47 lines
983 B
C++
|
|
// Copyright (c) 2015-16 Tom Deakin, Simon McIntosh-Smith,
|
|
// University of Bristol HPC
|
|
//
|
|
// For full license terms please see the LICENSE file distributed with this
|
|
// source code
|
|
|
|
#pragma once
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
// Array values
|
|
#define startA (0.1)
|
|
#define startB (0.2)
|
|
#define startC (0.0)
|
|
#define startScalar (0.4)
|
|
|
|
template <class T>
|
|
class Stream
|
|
{
|
|
public:
|
|
|
|
virtual ~Stream(){}
|
|
|
|
// Kernels
|
|
// These must be blocking calls
|
|
virtual void copy() = 0;
|
|
virtual void mul() = 0;
|
|
virtual void add() = 0;
|
|
virtual void triad() = 0;
|
|
virtual void nstream() = 0;
|
|
virtual T dot() = 0;
|
|
|
|
// Copy memory between host and device
|
|
virtual void init_arrays(T initA, T initB, T initC) = 0;
|
|
virtual void read_arrays(std::vector<T>& a, std::vector<T>& b, std::vector<T>& c) = 0;
|
|
|
|
};
|
|
|
|
|
|
// Implementation specific device functions
|
|
void listDevices(void);
|
|
std::string getDeviceName(const int);
|
|
std::string getDeviceDriver(const int);
|
|
|