mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-04-19 22:39:03 +00:00
ck-builder: tensor input/output reflection (#3536)
This adds some utilities to automatically generate UniqueInputs, UniqueOutputs, alloc_inputs, alloc_outputs, and validate, based on a Inputs::reflect() and Outputs::reflect().
This commit is contained in:
@@ -204,6 +204,7 @@ struct DummySignature
|
||||
constexpr DummySignature DUMMY_SIGNATURE = {};
|
||||
|
||||
namespace ck_tile::builder::test {
|
||||
|
||||
template <>
|
||||
struct Args<DUMMY_SIGNATURE>
|
||||
{
|
||||
@@ -225,6 +226,7 @@ struct Outputs<DUMMY_SIGNATURE>
|
||||
void* b;
|
||||
};
|
||||
|
||||
// Explicitly implement validate for this type to test that that works.
|
||||
template <>
|
||||
ValidationReport validate<DUMMY_SIGNATURE>(const Args<DUMMY_SIGNATURE>& args,
|
||||
Outputs<DUMMY_SIGNATURE> actual,
|
||||
|
||||
Reference in New Issue
Block a user