Skip to content
Snippets Groups Projects
Commit 85887601 authored by Hkorb's avatar Hkorb
Browse files

Adapt kernels to new StreamManager

parent 57577c32
No related branches found
No related tags found
1 merge request!170Kernel templetization and efficiency improvements
......@@ -13,7 +13,7 @@ class Kernel
public:
virtual ~Kernel() = default;
virtual void run() = 0;
virtual void runOnIndices(const unsigned int *indices, unsigned int size_indices, int stream = -1) = 0; //if stream == -1: run on default stream
virtual void runOnIndices(const unsigned int *indices, unsigned int size_indices) = 0; //if stream == -1: run on default stream
virtual bool checkParameter() = 0;
virtual std::vector<PreProcessorType> getPreProcessorTypes() = 0;
......
......@@ -3,7 +3,7 @@
#include "Kernel/Utilities/CheckParameterStrategy/CheckParameterStrategy.h"
void KernelImp::runOnIndices(const unsigned int *indices, unsigned int size_indices, int stream)
void KernelImp::runOnIndices(const unsigned int *indices, unsigned int size_indices)
{
printf("Method not implemented for this Kernel \n");
}
......
......@@ -14,7 +14,7 @@ class KernelImp : public Kernel
{
public:
virtual void run() = 0;
virtual void runOnIndices(const unsigned int *indices, unsigned int size_indices, int stream = -1);
virtual void runOnIndices(const unsigned int *indices, unsigned int size_indices);
bool checkParameter();
std::vector<PreProcessorType> getPreProcessorTypes();
......
......@@ -30,7 +30,7 @@ void CumulantK17CompChimRedesigned::run()
getLastCudaError("LB_Kernel_CumulantK17CompChim execution failed");
}
void CumulantK17CompChimRedesigned::runOnIndices(const unsigned int *indices, unsigned int size_indices, int streamIndex)
void CumulantK17CompChimRedesigned::runOnIndices(const unsigned int *indices, unsigned int size_indices)
{
cudaStream_t stream = para->getStreamManager()->getStream(CudaStreamManager::Stream::Bulk);
......
......@@ -8,7 +8,7 @@ class CumulantK17CompChimRedesigned : public KernelImp
public:
static std::shared_ptr<CumulantK17CompChimRedesigned> getNewInstance(std::shared_ptr<Parameter> para, int level);
void run() override;
void runOnIndices(const unsigned int *indices, unsigned int size_indices, int stream = -1) override;
void runOnIndices(const unsigned int *indices, unsigned int size_indices) override;
private:
CumulantK17CompChimRedesigned();
......
......@@ -30,7 +30,7 @@ void CumulantK17CompChimStream::run()
getLastCudaError("LB_Kernel_CumulantK17CompChim execution failed");
}
void CumulantK17CompChimStream::runOnIndices(const unsigned int *indices, unsigned int size_indices, int streamIndex)
void CumulantK17CompChimStream::runOnIndices(const unsigned int *indices, unsigned int size_indices)
{
cudaStream_t stream = para->getStreamManager()->getStream(CudaStreamManager::Stream::Bulk);
......
......@@ -8,7 +8,7 @@ class CumulantK17CompChimStream : public KernelImp
public:
static std::shared_ptr<CumulantK17CompChimStream> getNewInstance(std::shared_ptr<Parameter> para, int level);
void run() override;
void runOnIndices(const unsigned int *indices, unsigned int size_indices, int stream = -1) override;
void runOnIndices(const unsigned int *indices, unsigned int size_indices) override;
private:
CumulantK17CompChimStream();
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment