Kernels:

kernels-staging
/

flash-attn3

Trusted publisher

Kernel card Files Files and versions

flash-attn3 / benchmarks /benchmark.py

danieldk's picture

danieldk HF Staff

Uploaded using `kernel-builder`.

467200e verified about 1 month ago

317 Bytes

	from kernels.benchmarks import (
	FlashAttentionBenchmark,
	FlashAttentionCausalBenchmark,
	FlashAttentionVarlenBenchmark,
	)


	class FlashAttn(FlashAttentionBenchmark):
	pass


	class FlashAttnCausal(FlashAttentionCausalBenchmark):
	pass


	class FlashAttnVarlen(FlashAttentionVarlenBenchmark):
	pass