drbh HF Staff

feat: add benchmark file

dd401a6 verified 10 months ago

317 Bytes

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


class FlashAttn(FlashAttentionBenchmark):
    pass


class FlashAttnCausal(FlashAttentionCausalBenchmark):
    pass


class FlashAttnVarlen(FlashAttentionVarlenBenchmark):
    pass

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


	class FlashAttn(FlashAttentionBenchmark):
	pass


	class FlashAttnCausal(FlashAttentionCausalBenchmark):
	pass


	class FlashAttnVarlen(FlashAttentionVarlenBenchmark):
	pass