diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h')
| -rw-r--r-- | src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h b/src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h index c85ddf5a2c..328e942955 100644 --- a/src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h +++ b/src/dynamic_fusion/sketch/gpu/template_writer/IGpuTemplateComponentWriter.h @@ -48,6 +48,9 @@ class IGpuTemplateComponentWriter public: using ComponentGroup = GpuKernelComponentGroup; + /**For now all kernel intermeditate/destination tensors are expected to be of type Tensor_4D_t_Buffer*/ + static constexpr GpuKernelArgumentInfo::Type common_tensor_type = GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer; + public: /** Constructor * |