foreman
72 строки
1.8 KiB
C++
72 строки
1.8 KiB
C++
//
|
|
// Copyright (c) 2011 Advanced Micro Devices, Inc. All rights reserved.
|
|
//
|
|
|
|
#ifndef CPUKERNEL_HPP_
|
|
#define CPUKERNEL_HPP_
|
|
|
|
#include "top.hpp"
|
|
#include "device/device.hpp"
|
|
#include <amdocl/cl_kernel.h>
|
|
|
|
//! \namespace cpu CPU Device Implementation
|
|
namespace cpu {
|
|
|
|
//! \class CPU kernel
|
|
class Kernel : public device::Kernel
|
|
{
|
|
private:
|
|
const void* entryPoint_; //!< entry for the kernel
|
|
|
|
std::vector< std::pair<size_t, size_t> > args_;
|
|
public:
|
|
uint nature_; //!< kernel's nature
|
|
uint privateSize_; //!< WorkItem's private memory size (in bytes)
|
|
|
|
private:
|
|
//! Disable default copy constructor
|
|
Kernel(const Kernel&);
|
|
//! Disable operator=
|
|
Kernel& operator=(const Kernel&);
|
|
|
|
public:
|
|
void addArg(size_t size, size_t alignment) {
|
|
args_.push_back(std::pair<size_t, size_t>(size, alignment));
|
|
}
|
|
|
|
size_t getArgSize(int argIndex) const {
|
|
return args_[argIndex].first;
|
|
}
|
|
|
|
size_t getArgAlignment(int argIndex) const {
|
|
return args_[argIndex].second;
|
|
}
|
|
|
|
//! Default constructor
|
|
Kernel(const std::string& name)
|
|
: device::Kernel(name), entryPoint_(NULL), nature_(0),
|
|
privateSize_(CLK_PRIVATE_MEMORY_SIZE)
|
|
{
|
|
workGroupInfo_.size_ = CPU_MAX_WORKGROUP_SIZE;
|
|
}
|
|
|
|
//! Default destructor
|
|
~Kernel() {}
|
|
|
|
//! Returns the CPU kernel entry point
|
|
const void* getEntryPoint() const { return entryPoint_; }
|
|
|
|
//! Sets the CPU kernel entry point
|
|
void setEntryPoint(const void* entryPoint) { entryPoint_ = entryPoint; }
|
|
|
|
//! Returns true if the kernel has a call to barrier
|
|
bool hasBarrier() const { return 0 != (nature_ & KN_HAS_BARRIER); }
|
|
|
|
//! Returns the private memory size of a single WorkItem
|
|
uint getWorkItemPrivateMemSize() const { return privateSize_; }
|
|
};
|
|
|
|
} // namespace cpu
|
|
|
|
#endif // CPUKERNEL_HPP_
|