alpa/__init__.py,sha256=0r3FkXxK_MrWi88b-vgoUQlK5Ly1sA1X8AJJtIQTsP4,1509
alpa/api.py,sha256=QgLcj9JpckyElYuJJx1Rbx_M2Rc0pJQTPYZOUWRm23o,8946
alpa/data_loader.py,sha256=De5UP8ZZzUdZxoUjq13ddTl8wknA8sGpiB-0b5_xuj8,9511
alpa/device_mesh.py,sha256=g7DUISVNiK6_UZ2XODUFpRvU_RIG0ndTEJx3buzWMGA,90321
alpa/global_env.py,sha256=vKiH9LGJIOKmME4TvIYPcIO-SMT0UXFurh1UxWL_6m8,3601
alpa/measure_record.py,sha256=Vzaub8P4YZMtrZCfeV8dUo-0LHW9w8tGhtzmHk8Y0FY,5205
alpa/mesh_executable.py,sha256=VtABU7MQn-mA29Rm_1DbtdlU69yOSdVzGRIONCvGbD8,52012
alpa/mesh_profiling.py,sha256=9QVPtSkbpdor-o64STqv9WraTU3OGFifKaudd6bcSMc,33832
alpa/monkey_patch.py,sha256=BPMdzHVmSqf7ppfy8oqmtFr8kllyLmnNibKYyGVISqE,11341
alpa/parallel_method.py,sha256=k91S2dk3OR72Yhpkb7Qxx5fGp_oW9qiNcPd3vHFePVk,10154
alpa/serialization.py,sha256=SS6QWihw28i12eTWY-zI4drxzmCbQoS8KYVs7Grh-yo,7852
alpa/testing.py,sha256=jkhJn_y-yltHYyXx2qZDnH9DJxb2mOb-HlVt_ylbxPo,14788
alpa/timer.py,sha256=kY0UgFhDIA-no_FcWVRqT_XdGWOjubfQrpBOjwasNvY,3173
alpa/util.py,sha256=GZgKfTOfbF0c0IRRIZBi9tkZQOMbIgnMYTRo0vxpTbE,35703
alpa/collective/__init__.py,sha256=k_PxOhEsYesymIUKjyKYrsywdOmaZwqSICkLSQrArfw,1019
alpa/collective/collective.py,sha256=100QMLzIGHDvqCkuMdHJxdWm5QDAuSEyFcAC0evSxVE,27865
alpa/collective/const.py,sha256=r0h9zZAnN-vsibpUwohoaU9XMI1xDkJq9DcW4Fj829M,891
alpa/collective/types.py,sha256=cFj0rrD4HG13Xw9ytrH5vq-2CU6Pl9aHU59Mn6K_7uc,2138
alpa/collective/util.py,sha256=Bmr8mNdfyvIgQqpNYt5xrN7e_pUNQv1zYI78DhtVTww,2104
alpa/collective/collective_group/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
alpa/collective/collective_group/base_collective_group.py,sha256=x5Cmeqj7OvVCuGmgnUrNesi8anfydddifaDaIb9Lo54,2427
alpa/collective/collective_group/cuda_stream.py,sha256=YRHH4kPyCu_TXrpAnzy5IzUcBNvU0LZ61yQ_IgtlYyU,2955
alpa/collective/collective_group/gloo_collective_group.py,sha256=eYsfKUoKE63f1yjZQNfG8YEbFIByStGyiUt8BGnJaO4,19438
alpa/collective/collective_group/gloo_util.py,sha256=asG3WDA7Gv17TVTK4sR55Li2lbfOoaPAvtnlXMwABfA,8483
alpa/collective/collective_group/nccl_collective_group.py,sha256=wZEvvXGoV41kqgX6RIPq3fS03eMTrr6tmpB29BUzyJc,37881
alpa/collective/collective_group/nccl_util.py,sha256=HDwDu27qRQQtIZ7KrxPROCedb_yn-lcrvQbJSZO7qO0,9124
alpa/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
alpa/model/bert_model.py,sha256=S3_IEcHWFztZU8ivj6R8p3qqUgxeS0nprH2rkGG5I58,29951
alpa/model/conformer.py,sha256=5JwV0LezyBfUCNCswY6NeRf-X04HrhO2wEfT_NZwoQg,12248
alpa/model/gpt_model.py,sha256=wAqnFN5XfvM6vittss0VtmF14cFOpmBLAhE0aWijnrU,5206
alpa/model/model_util.py,sha256=w6Ny4-opSzUuMggKMUN6BG9N1IH2iTZCg9yFsZp5dQM,17427
alpa/model/moe.py,sha256=JiCDMd_9-VrTuautMRA3Er1hca3WHFsquvQsYzWHqOQ,15744
alpa/model/wide_resnet.py,sha256=_GHy6hFKEJwnF3gu_qw8Kh6h0aar6O4Bq4yzNYjjmk0,5272
alpa/pipeline_parallel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
alpa/pipeline_parallel/apply_grad.py,sha256=J_w0MjpNT2usbwOcmXh-jXSze43_B4Kf0lBoSWL7r1o,32734
alpa/pipeline_parallel/compile_executable.py,sha256=wMlNjYzljX0f0dDn6Yu_REkfLIiwDYuVuQm-mT8BCzc,18131
alpa/pipeline_parallel/computation.py,sha256=apIasMSJCqxk9aS0yiL8EIopm_GgVcMimEgyhr35IuY,51889
alpa/pipeline_parallel/cross_mesh_resharding.py,sha256=orPsBo6myHpMgQzCNWHAnbi8a0S8IMBgThxCF3eQYH4,67103
alpa/pipeline_parallel/layer_construction.py,sha256=EQh2UQS55sucJfF6ARXJZJ_1SOmIF1jlH4cWm5iJSj4,22631
alpa/pipeline_parallel/layer_stats.py,sha256=XDHeujarFhQNEr0-prEsBb6XcUdeNxK12mWGlontJbE,5547
alpa/pipeline_parallel/local_pipeline.py,sha256=YvIgvD_Yx8VAnEmV9R7nTF_BFsPoZXKElbqoiCcQkHk,5627
alpa/pipeline_parallel/pipeshard_executable.py,sha256=UU69k29yhJfyfduGe8AgtmvuPxklXPN1zf32KKQDaKY,24342
alpa/pipeline_parallel/primitive_def.py,sha256=lztBB4vw-AyEPaBXZY6n4t6Kb3J-yuYrCGbtsEN_Uvg,6378
alpa/pipeline_parallel/resharding_tensor.py,sha256=WtlvoSCGyDfAwwDyhEgB_5Pk_to6P1ZAhjVYl-gRCu8,9179
alpa/pipeline_parallel/runtime_emitter.py,sha256=VXnooX7Am1uN9NVpoeo14KPSGMlPbyaJfKeP4nfZztU,49646
alpa/pipeline_parallel/schedules.py,sha256=eanbMIu-dpuMpYbSWnql9EQYxaTuFxTCFgDOs8O7uEE,15442
alpa/pipeline_parallel/stage_construction.py,sha256=SsNRcPrvxu0ejCIFCk-UgcoFOXcyyya3gUKYa7LJeUI,33623
alpa/pipeline_parallel/stage_profiling.py,sha256=o1tJAzkDj3sbN0FyR45AFtXMnpKPsZlcmvZByvZ84GE,36869
alpa/pipeline_parallel/xla_custom_call_marker/__init__.py,sha256=v9G7S-jP2iYL3a_lLQbMr_qFPEvyzCIGridKvGDO6B8,340
alpa/pipeline_parallel/xla_custom_call_marker/build/xla_custom_call_marker.cpython-38-x86_64-linux-gnu.so,sha256=yqFXJcccU4xMmg_8gcpRIENGrU5hubJaTCsbWRPpGIg,1300400
alpa/shard_parallel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
alpa/shard_parallel/auto_sharding.py,sha256=FRXxIh3theGOUVaJzLetarKeE1rh5t0sT8pgc9CS3AE,33119
alpa/shard_parallel/compile_executable.py,sha256=uNUpAzp4kOWaYyVac81zaL66hHWg7XajY6oFC1Sw4ug,16033
alpa/torch/__init__.py,sha256=zTO-2MpbkSyMw93hRXvfUMzJhMtXFPkDoQH-xIv3nnI,5204
alpa/torch/tensor_utils.py,sha256=LiNMlwL7MIjQwxlZSHORp992fR5qjazUQ3riRPMQnlY,4190
alpa/torch/trainer.py,sha256=pZ52Lliolt9pDXVjk2X6S87JTOaZkts-J4-Kf7vZFkc,6130
alpa/torch/nn/__init__.py,sha256=7DICK0t6XVAs1DV59AtdatNvE6upw827HNUGpqcpHnQ,20192
alpa/torch/nn/utils.py,sha256=mWb2iTxoEf39dct9xjIQg7GgH5iDDgb1yG-7Pl571pA,17130
alpa/torch/ops/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
alpa/torch/ops/mapping.py,sha256=vK0ZpI89lxe-ezhqydsdDTltOAtjJQcpyuCUFzYP01E,27899
alpa/torch/optim/__init__.py,sha256=Qh74GFhfDvZHSdc4XTV8oCwzK4auRqbEyXp-Ajf_1M8,41
alpa/torch/optim/adam.py,sha256=Z9gwIZBBZf6ULeP-acnl_uun4JECixm5suCA-JgaOiw,1518
tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
tests/run_all.py,sha256=MadeQ8JcItlLUE-J-KX2Td0jwMN3xULTgJWwwoblUso,3037
tests/test_auto_sharding_basic.py,sha256=AcScCR1StpiODsTT_9tKmHUeB7XEqIJzivGDEf1HF94,7265
tests/test_auto_sharding_bert.py,sha256=K3KDYnhh_1_6Xi061laEd2bMqqeJt78Gqgo3JQneteU,27208
tests/test_auto_sharding_conv.py,sha256=2vLeUWP76irmFCxwT6LI1nW0ly8gB22nv6mXNZA5Gsk,11189
tests/test_auto_sharding_gradient_accumulation.py,sha256=9vZ8WccdhCAwPJQGeQTPvgSmL-nWFS8glOAY2cyPUPw,5816
tests/test_auto_sharding_mixed_2d.py,sha256=jttMSW9ZipEAx9C9ZXdJkAn-GyrYjrOFGkT2YJKeGhk,3391
tests/test_auto_sharding_mlp.py,sha256=2qG_zP3ofjoTJSVyauJJO5JYqqlwDJowNjjXFPQLPdk,17283
tests/test_auto_sharding_moe.py,sha256=ud_GP_rmOdRFyPpY-Rh-oEuJOA73EtV8Vg1MXCUPf_w,16321
tests/test_auto_sharding_numerical.py,sha256=H7OHS-ZlAkfUXKcrvf4a5AKzdgYK6rr-1Zh4ozBiGTU,4318
tests/test_cross_mesh_resharding.py,sha256=3n_UZYbsRzs6sBF2Oa7aUkAyH0edLygwp7FGQIxW41Y,15897
tests/test_data_loader.py,sha256=GCyjMzwqpW1MGdDGSdkycqFKyTB5krOTpiCparFQSQ8,3033
tests/test_device_mesh.py,sha256=Cfy9235gwgSZYc-EZKNurbLIlvXoLicm3gElhqv-BFQ,1955
tests/test_dist_save_load.py,sha256=VAiyBh7YHPqJkfju5mCHLIC5BNioqZFXBPrSasti8AU,12109
tests/test_dynamic_programming.py,sha256=fSkS8BMmx3bfIbI2T1Q31CG3OAP-yA-Hc4pgMVaUqCc,2553
tests/test_hlo_cost_model.py,sha256=f9isKoTqpatPZtiIHW3GB9F_U3e5sYt1oo9TaryNrxg,3538
tests/test_install.py,sha256=5Dz4LDPnGQ5dFOuejz4vH68PKXKvpOMkaCO6b2H0ZgU,3676
tests/test_memory_leak.py,sha256=sNzGUUMkFXeKUnPJHhkfYojskdV4-ys-mKFrHQO1XvU,2680
tests/test_ordered_set.py,sha256=duJIeW0JXn-1BPwwbwhUA3nVGMEFeHdHD1rJbqeKGj8,3532
tests/test_pipeline_bert.py,sha256=6myYUZhg1REB46c0aa38NsshFnEFNGZsI6K3psmOq9o,3148
tests/test_pipeline_inference_only.py,sha256=M0gWmDTm-qRfRja27aCA4pWg01B8YvBF79I_hTU3aWU,4136
tests/test_pipeline_layer_construction.py,sha256=AyKc6ddAiwxlFwCbxF5H3WjnmNT-KzTn_aoE0OK6vP4,924
tests/test_pipeline_marker.py,sha256=CUBEf9YMSAJW9Wji9b21uHe-Bscb2BNOeEkM3wAw7Jo,2799
tests/test_pipeline_mlp.py,sha256=1sWIANP2FE5pA3rX_l0nwCHlKPTb-H0tTThyxR_k6fg,2585
tests/test_pipeline_multiple_graph.py,sha256=U-qgzg_GwMbEcglZxYaXK1d7kOYmZ5q5pDRaPpCU3VA,2575
tests/test_pipeline_numerical.py,sha256=w7rNvfBo8f8e58bwLf2uD1v3ELZ1iheQkfPUzn7xEC0,1727
tests/test_pipeline_reduce_scatter.py,sha256=4yzflJswn-CjzLLHNyzxI_o--JLgfUmQ44hVGMc9yZo,3998
tests/test_pipeline_remat.py,sha256=1IrAOq2jEM_Ie2wLLfUSfJya2gaFNvrMQTPZWP7Y3Y8,1231
tests/test_pipeline_scatter_gather.py,sha256=ZzkIONIvUJsydTCr8Vj9yECpUg1Uq5S_arMeP2_kC-Q,1264
tests/test_pipeline_schedules.py,sha256=ezaUAj3yfXPKzLiaG4Fyh9_URrlPOi1SueW73ua3wek,4269
tests/test_pipeline_stage_construct_util.py,sha256=K3f30plX1M9QBpNCyPuyA-NZqAOmDiCRevLJLYKhOkA,14398
tests/test_pipeline_stage_construction.py,sha256=Ab9anKC7U1sttgSctAsrIZUb360NGqY262jnmUu7E7o,1806
tests/test_pipeline_tied_embedding.py,sha256=BmVY3GTCJWoXADkpr9sY9BLmJmkWUNZZklV-r_diLMs,2739
tests/test_save_load.py,sha256=rqic0ZjaG0D3dboGu3QdKnzNlTrE5TxaGxGRbbfE_Zs,3211
tests/test_torch_dict_input.py,sha256=99OPPZ5U9ut7ie5x3AzeKSNSMzNcKHsLekfMyHzjhEI,2237
tests/test_torch_reshape.py,sha256=q56sxWQVdcQKsexNfPadrM3hJ4KHUD-j5t-mxLqwGvI,1740
tests/test_torch_simple.py,sha256=0nVKMBbtZ-2PePlMwCIizMKHntx7ovvHmVXwH6y5gV4,2621
tests/test_torch_zhen.py,sha256=SdSd2IWhGYmfaSvBGhyYYjLyvPgYm72FwZOpgJ3nkyw,21375
alpa-0.0.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
alpa-0.0.3.dist-info/METADATA,sha256=6cCUiTXQqG6qcEIsR8Wa8iyyOvl9HUPOQ0DymdBzJEc,4244
alpa-0.0.3.dist-info/WHEEL,sha256=HDpOoUm88OZsXwJ1fG8bFxqAdtiJNfIcxp3_RuQe18I,103
alpa-0.0.3.dist-info/top_level.txt,sha256=Sn97Y8lTDhV8CAyicYWuxUvUzborH1aYhZ3Wi_LWr-o,11
alpa-0.0.3.dist-info/RECORD,,
