p32581460
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add: comments for Kernel

This commit is contained in:
Liyan Zheng 2022-08-09 20:05:01 +08:00
parent ce5d49c79b
commit 8219b0f7ff
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
include/core/kernel.h
View File

@ -14,11 +14,19 @@ class Kernel {
Kernel() {}
virtual ~Kernel() {}
/**
* @param op The operator to be executed.
* @param record The parameters for kernel execution. If extra parameters
* are required, inherit from PerfRecord and add extra parameters.
* Otherwire, use PerfRecord directly.
*/
virtual void compute(const Operator &op,
const PerfRecord &record) const = 0;
// This function call compute with a default record.
/**
* @brief Executes an op with a default parameter.
*/
virtual void compute(const Operator &op) const = 0;
// Tuning should be idempotent since it is called multiple times.
// Premise: op is idempotent since it is called multiple times.
virtual PerfRecord tune(const Operator &op) const = 0;
};
@ -55,6 +63,8 @@ class KernelRegistry {
}
};
} // namespace infini
#define _REGISTER_KERNEL_1(device, opType, dataType, kernel, name, cnt) \
namespace infini { \
static const bool _CAT(_register_kernel_, cnt) = \
@ -64,5 +74,3 @@ class KernelRegistry {
#define REGISTER_KERNEL(device, opType, dataType, kernel, name) \
_REGISTER_KERNEL_1(device, opType, dataType, kernel, name, __COUNTER__)
} // namespace infini