flash_fwd_kernel.h

/******************************************************************************************
 * Copyright (c) 2025, Baohui.Fang, Yushun.Zhang, Chang.liu, Wenjian.Zhang, Jianbang.Xu
 *****************************************************************************************/

#pragma once
#include "flash_fwd_b16_fa.h"
#include "flash_fwd_b16_pa.h"
#include "flash_fwd_b16_mla.h"
#include "flash_fwd_b8_fa.h"
#include "flash_fwd_b8_pa.h"
#include "flash_fwd_b8_mla.h"
#include "flash_fwd_b4_fa.h"
#include "flash_fwd_b4_pa.h"
#include "flash_fwd_b4_mla.h"
#include "flash_fwd_reduce.h"