o vžÄi;ã@sšdZddlmZddlmZ         ddededed ed edBd edBd edBd edBdedBdedBdededBdee eefBfdd„Z dS)z» Imaginaire4 Attention Subpackage: Unified implementation for all Attention implementations. cuDNN Backend: intermediate API stubs Always safe to import (as long as torch is available.) é)ÚTensor)Ú CausalTypeFNÚqueryÚkeyÚvalueÚ is_causalÚ causal_typeÚscaleÚcumulative_seqlen_QÚcumulative_seqlen_KVÚ max_seqlen_QÚ max_seqlen_KVÚ return_lseÚbackend_kwargsÚreturnc Cstdƒ‚)NzrTried to run cuDNN attention, but it is not supported / available. Try running with debug logs enabled to see why.)Ú RuntimeError) rrrrrr r r r r rr©rúY/data/cameron/vidgen/cosmos-policy/cosmos_policy/_src/imaginaire/attention/cudnn/stubs.pyÚcudnn_attentionsÿr) FNNNNNNFN) Ú__doc__ÚtorchrÚ-cosmos_policy._src.imaginaire.attention.masksrÚboolÚfloatÚintÚdictÚtuplerrrrrÚsN  ôÿþýüûúùø ÷ ö õ ô ó