o Ô°“i¡ã@s^ddlmZmZddlmZmZddlmZmZddl m Z m Z Gdd„dƒZ e ƒZ dgZdS) é)ÚSelectionAttentionÚselection_attention_wrapper)ÚCompressionAttentionÚcompression_attention_wrapper)ÚSlidingWindowAttentionÚ sliding_window_attention_wrapper)Ú TopKReductionÚtopk_reduction_wrapperc@sLeZdZeeƒZeeƒZeeƒZeeƒZeeƒZee ƒZ ee ƒZ ee ƒZ dS)Ú NSANamespaceN) Ú__name__Ú __module__Ú __qualname__Ú staticmethodrrrrrrrr ©rrúZ/home/ubuntu/.local/lib/python3.10/site-packages/cudnn/native_sparse_attention/__init__.pyr s r ÚNSAN)Ú selectionrrÚ compressionrrÚsliding_window_attentionrrÚtop_krr r rÚ__all__rrrrÚsÿ