flash-attention/hopper/copy_paged_sm90_tma.hpp

#pragma once
#include <cutlass/version.h>

#if CUTLASS_VERSION >= 360
#include "copy_paged_sm90_tma_cutlass36.hpp"
#else 
#include "copy_paged_sm90_tma_cutlass35.hpp"
#endif
Paged Attention support for FA3 (#1268) 2024-11-10 09:05:01 +08:00			`#pragma once`
Make FA3 paged attention ready for upgrade to Cutlass 3.6 (#1331) 2024-11-13 03:31:37 +08:00			`#include <cutlass/version.h>`
Paged Attention support for FA3 (#1268) 2024-11-10 09:05:01 +08:00
Make FA3 paged attention ready for upgrade to Cutlass 3.6 (#1331) 2024-11-13 03:31:37 +08:00			`#if CUTLASS_VERSION >= 360`
			`#include "copy_paged_sm90_tma_cutlass36.hpp"`
			`#else`
			`#include "copy_paged_sm90_tma_cutlass35.hpp"`
			`#endif`