sh06SSKrSSKrSSKrSSKJr SSKJrJr SSKJ r SSK J r SSK J r Jr SSKJrJrJr SSKJr SS KJrJr S rS rS rS rSr/SQr\R<R?SS\R@RBS9 \R<RESS5S5r#\R<RESS5S5r$\R<RKS5S5r&Sr'Sr(Sr)\S5r*Sr+Sr,Sr-Sr.S r/S!r0S"r1S#r2\"5r3S$r4S%r5\"S&S'\2\1\ "\Rl\Rn5S(S(S(S(S(S(S(S)9 \"S*S'\+\)\ "\Rl\Rn5S(S(S(S(S(S+S(S)9 \"S,S'\\\ "\Rl\Rn5S(S(S(S(S(S-9 \"S,S.\\\ "\Rl\Rn5S(S(S(S(S(S-9 \"S,S/\\\ "\Rl\Rn5S(S(S(S(S(S-9 \"S0S'\(\'\ "\Rl\Rn5S(S(S(S(S(S+S(S)9 \"S1S'\4\)\ "\Rl\Rn5S(S(S(S(S(S+\"\RpS2S35\"\RpS2S45\"\RpS2S55\"\RpS2S654S(S79 \"S8S'\5\)\ "\Rl\Rn5S(S(S(S(S(S+S(S)9 \"S9S'\0\/\ "\Rl\Rn5S(S(S(S(S(S(S:9 \"S;S'\-\,\ "\Rl\Rn5S(S(S(S(S(S(S:9 \"SS'\\.\"\Rr\Rt5S(S(S(S(S(\"\RpS2S35\"\RpS2S45\"\RpS2S55\"\RpS2S654\ /S=9 / r;g)?Nmap)_create_empty_block_maskflex_attention) make_tensor)onlyCUDA) all_types_and custom_types) DecorateInfoOpInfo SampleInput)mark_compile_region) InvokeQuantinvoke_quant_packedc +# [R"[XUS9n[U"SSSSSS9U"SSSSSS9/U"SSSS9U"SSSS94S9v g7f)Ndevicedtype requires_grad皙?lowhigh)args functoolspartialrr opinforrrkwargsmake_args r/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/testing/_internal/hop_db.pysample_inputs_mapr%sg  F}H  !Qs +XaA3Q-OPqc*HQCa,H I sAA cUSR5RS5U-USUR5-R5R USR 55/$)Nrg?r)cosadd_sincos_viewsizexy0y1s r$inner_fr1sQ aDHHJOOC 2 %!rvvx'='='?'D'DQqTYY['Q RRc Sn[X0X5$)Nc[XU5$Nr1r-s r$fsimple_map..f sqb!!r2r)xsr/r0r7s r$ simple_mapr:s" qb r2c Sn[X0X5$)Nc Sn[X0X5$)Nc[XU5$r5r6r-s r$f2"nested_map..f1..f2(s1"% %r2rxxr/r0r>s r$f1nested_map..f1's &22""r2rr9r/r0rBs r$ nested_maprE&s# rr r2c Sn[X0X5$)Nc Sn[X0X5$)Nc Sn[X0X5$)Nc[XU5$r5r6r-s r$r>5triple_nested_map..f0..f1..f23sqb))r2rr@s r$rB)triple_nested_map..f0..f12s *rr& &r2rrDs r$f0triple_nested_map..f01s ' 22""r2r)r9r/r0rLs r$triple_nested_maprN0s# rr r2)custom_function_callautograd_function_applyrun_and_save_rng_staterun_with_rng_stategraphsafe_run_with_rng_state out_dtype trace_wrappedtag_activation_checkpointexecutorch_call_delegatewrapwrap_with_set_grad_enabledauto_functionalized_v2associative_scan flat_applywrap_with_autocastwrap_activation_checkpointrun_const_graphauto_functionalizedrmap_impl with_effects strict_mode_export_tracepointcall_torchbindtriton_kernel_wrapper_mutation triton_kernel_wrapper_functional hints_wrapper foreach_mapaoti_call_delegateztestlib::mutating_custom_opz8(Tensor(a!) x, Tensor(b!) z) -> (Tensor, Tensor, Tensor))tagscpucRURS5 URS5 XX-4$Nr(r.zs r$ foo_impl_cpursr$FF1IFF1I ;r2cudacRURS5 URS5 XX-4$rnrprqs r$ foo_impl_cudarwyrtr2cXX-4$r5rqs r$foo_impl_abstractrzs ;r2c +j# [R"[XUS9n[U"SSSSSS95v g7fNrrrrrr s r$sample_inputs_condr}6  F}H hq!QCa8 9913c\[R"UR5S:SSU/5$)Nrc$UR54$r5r'r.s r$simple_cond..s aeegZr2c$UR54$r5)r)rs r$rrs AEEG:r2)torchcondsumrs r$ simple_condrs& ::aeegk#79MPQs SSr2c +j# [R"[XUS9n[U"SSSSSS95v g7fr|rr s r$sample_inputs_invoke_subgraphrr~rc.[R"U5$r5rr)rs r$fn_for_invoke_subgraphrs 99Q<r2c[U5$r5)rrs r$simple_invoke_subgraphrs !! $$r2c +|# [R"[XSS9n[U"SSSSSS9U"SSSSSS95v g7f)NFrrrrrr s r$ sample_inputs_auto_functionalizersJ  FuH Aqc*HQ1#A,N s:<cT[RRRX5$r5)ropstestlibmutating_custom_oprqs r$simple_auto_functionalizers 99   / / 55r2c+^ # [R"[XUS9m SnU 4Sj[S55upgn[ Xg5n [ XgXU 5v g7f)Nrc X-$r5ry)scorebhmns r$ score_mod/sample_inputs_flex_attention..score_mods yr2c 3:># UHnT"SSSSSSS9v M g7f)rrrNry).0_r#s r$ /sample_inputs_flex_attention..s I1x1c1#A6s)rrrrangerr ) r!rrrr"rqkv block_maskr#s @r$sample_inputs_flex_attentionrsP  F}HJaIGA!)!/J aA* 55sAAc +# [R"[XSS9n[[R "S5U"SSSSSS95v g7f)NFrrrrr)rrrr rtensorr s r$sample_inputs_while_looprsF  FuH  QAqc* sAAcPSnSn[RRX#X45$)Nc US:$)Nrryiter_tr.s r$cond_fn"simple_while_loop..cond_fns zr2c,US- UR54$)Nrrrs r$body_fn"simple_while_loop..body_fnsz1557""r2)r_higher_order_ops while_loop)rr.rrs r$simple_while_looprs)#  " " - -g LLr2c +z# [R"[XUS9n[U"SSSSS9U"SSSSSS95v g7fr|rr s r$sample_inputs_scanrsI  F}H A3Q'Aqc* s9;cHSn[RRX U5$)Nc2X-U-nX R54$r5)clone)carryr.results r$ combine_fnsimple_scan..combine_fnsQ{{}$$r2)rrscan)initr9rs r$ simple_scanrs"%  " " ' ' " ==r2c,Sn[XU5SS-$)Nc6[R"U5U-4$r5r)r.ys r$fnsimple_invoke_quant..fns ! q ""r2r@) quant_tracerr.rs r$simple_invoke_quantrs # q !! $r ))r2c*Sn[X5SS-$)Nc0[R"U54$r5rrs r$r&simple_invoke_quant_packed..fns ! r2rr)rrs r$simple_invoke_quant_packedrs r %a (2 --r2rsimpleF) namevariant_test_nameopsample_inputs_funcdtypes supports_outcheck_batched_gradcheck_batched_gradgradcheck_batched_forward_grad"check_inplace_batched_forward_gradsupports_autogradsupports_gradgradinvoke_subgraphTr) rrrrrrrrrrnested triple_nestedr invoke_quantTestHOPtest_aot_exporttest_pre_dispatch_exporttest_serialize_exporttest_retrace_export) rrrrrrrrrrrskipsrrr) rrrrrrrrrrrauto_functionalizer) rrrrrrrrrrr decoratorsflex_attention_backward)r s 3V%?LQQGDS 640@ !>  $ $159: 16:; :;<:T:%6 6M>} *. " -UZZ4 $#(+0 " !8UZZ4 $#(+0 " ,UZZ4 $#(+0  " ,UZZ4 $#(+0  ) ,UZZ4 $#(+0  " -UZZ4 $#(+0 " 8UZZ4 $#(+0 119>O P (()5O  119>U V 119>S T   -0 "" %8UZZ4 $#(+0 " 3UZZ4 $#(+0  !" $;UZZ4 $#(+0  " 7EMM5==9 $#(+0 119>O P (()5O  119>U V 119>S T  :'* &" 7EMM5==9 $#(+0 119>O P (()5O  119>U V 119>S T  :'Q} r2