rmsnorm2d_fwd_pipeline_model_sensitive_pass.hpp File Reference

rmsnorm2d_fwd_pipeline_model_sensitive_pass.hpp File Reference#

Composable Kernel: rmsnorm2d_fwd_pipeline_model_sensitive_pass.hpp File Reference
rmsnorm2d_fwd_pipeline_model_sensitive_pass.hpp File Reference
#include "ck_tile/core.hpp"
#include "ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include <string>
#include <type_traits>

Go to the source code of this file.

Classes

struct  ck_tile::Rmsnorm2dFwdPipelineModelSensitiveT5Pass< Problem_, Policy_ >
 This T5Pass implements the RMSNorm2d forward pipeline as a variant based on Rmsnorm2dFwdPipelineOnePass and Rmsnorm2dFwdPipelineTwoPass using a T5 model-like method. More...

Namespaces

namespace  ck_tile