o E‘Dg[ã@spddddddœZddlmZddlZddlmZddlmZmZmZm Z m Z d ded eee d efd d „Z dS)TF)Úepilogue_fusionÚ max_autotuneÚ shape_paddingz trace.enabledztriton.cudagraphsé)ÚTensorN)Ú functional)ÚFÚListÚOptionalÚTupleÚnnÚinputÚ output_sizeÚreturnc Csp|jdkr tdƒ‚t|jtƒsJ‚d}| |||j|j|j||j¡}t   ||j |j |j|j||j |j¡ |j¡S)NÚzerosz:Only `zeros` padding mode is supported for ConvTranspose1dé)Ú padding_modeÚ ValueErrorÚ isinstanceÚpaddingÚtupleÚ_output_paddingÚstrideÚ kernel_sizeÚdilationrÚconv_transpose1dÚweightÚbiasÚgroupsÚtoÚdtype)Úselfr r Únum_spatial_dimsÚoutput_padding©r#ú8/home/user/app/unsloth_compiled_cache/ConvTranspose1d.pyÚforwards þ þþr%)N) Útorch_compile_optionsÚtorchrÚtorch.nnrrÚ*transformers.models.mllama.modeling_mllamarr r r Úintr%r#r#r#r$Ús   $