alpa/__init__.py,sha256=MZmm7RF5MGlvYSW9vv3RRlJgEcoC4PzJeOvLh2biW4g,2045
alpa/api.py,sha256=_ejBewIur71rX6wbfdvBvsaajzZsxDU5Ijj595VgZ6g,9267
alpa/create_state_parallel.py,sha256=XvlOguwQ8wPuAYGC1oymWICFbNBiEGVne4aIOGhTPog,7668
alpa/data_loader.py,sha256=vZjXvq5YbijzE414EUfDEo2GkHEARizTXR-eLpV1a5I,9269
alpa/device_mesh.py,sha256=WGzGhI0W9t1KWBT7WpTecPrzZ71NPtXQ9U-E-CeCWCE,90637
alpa/global_env.py,sha256=huVIeJ7fFCZ7Zi58SS7Coyv8c6bp6EXKTu8fdqjpkBs,3848
alpa/mesh_executable.py,sha256=1Fm0PRDq4Kr_TS2FYjPrGsrhNZpvCb-GhmMW4XMSjIU,49033
alpa/mesh_profiling.py,sha256=doi4PYvG6lGR2rIN1GYI8z4qMauAHMwOewbDilSkqVk,33836
alpa/monkey_patch.py,sha256=BPMdzHVmSqf7ppfy8oqmtFr8kllyLmnNibKYyGVISqE,11341
alpa/parallel_method.py,sha256=GuFEACusioMSMsPhas2bgxvlO4xxxqD0nnlXiGGbAgA,11322
alpa/parallel_plan.py,sha256=cGRAmoYdqzkyv17aQBZIG0d2BkLaStNsgk444kaG4XI,1238
alpa/serialization.py,sha256=ByUC9-LcZWsJ7mZmjvWSA3OVBPcxUZcUKxoZkBLGVGw,7433
alpa/testing.py,sha256=5j1JmxDBNW9TpXgqwxEXQTCBoGsgPxWpwtvXibDoG_U,13418
alpa/timer.py,sha256=kY0UgFhDIA-no_FcWVRqT_XdGWOjubfQrpBOjwasNvY,3173
alpa/util.py,sha256=-vO0cJWCLSsDxHC38frcsbCIgb73Wy52aldc_8imgRg,36157
alpa/collective/__init__.py,sha256=k_PxOhEsYesymIUKjyKYrsywdOmaZwqSICkLSQrArfw,1019
alpa/collective/collective.py,sha256=pwXbQr87ysEAcQkgcgOmAiMn7ps05riYMrLIjcFMc5g,29420
alpa/collective/const.py,sha256=r0h9zZAnN-vsibpUwohoaU9XMI1xDkJq9DcW4Fj829M,891
alpa/collective/types.py,sha256=WcQii7V-BMsh2NqVXQ12vxkNFKf3-q7JwwvM-vkkUFA,2204
alpa/collective/util.py,sha256=Bmr8mNdfyvIgQqpNYt5xrN7e_pUNQv1zYI78DhtVTww,2104
alpa/collective/worker_nccl_util_cupy.py,sha256=7fXMmTj6T2ESV0awq_ffO0mwyn3064VNHAgae5Irx6s,11586
alpa/collective/worker_nccl_util_xla.py,sha256=0seIX6pLwMJ6-dYy6Kn6ezQcGmMziEYAqMXps4O3MLI,7403
alpa/collective/collective_group/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
alpa/collective/collective_group/base_collective_group.py,sha256=oYnoEW4XJ5mYXhYtQ7MgbdXIiXjihFsgkkDAQQfDKrg,6231
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=byM8j3PC_lDWFjImDMVj5hpJmRhhVkZnMbtZfye9nUg,34203
alpa/collective/collective_group/nccl_util.py,sha256=HDwDu27qRQQtIZ7KrxPROCedb_yn-lcrvQbJSZO7qO0,9124
alpa/collective/collective_group/xla_nccl_collective_group.py,sha256=PyKQ2p1lVE1qBgFrvsWP8h9y-Mgsftni-dz7MLWt9z8,15677
alpa/collective/collective_group/xla_nccl_util.py,sha256=NrVBRKLiFTwg6p_FtxTBbCMKzzBiSgD191IX7qNeANc,229
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=fBtxQCkSGvQlG-Sv2iEe62QLfSyMsIlR2GeUmDsnk-4,33082
alpa/pipeline_parallel/compile_executable.py,sha256=P-PUTITt2dXL_KyPoFYepQaPzUSJXsGYW93kU3PjDm8,20089
alpa/pipeline_parallel/computation.py,sha256=jnUfI-McI4ASkMB3OBSpDqewQAYDj3_SE36d1LKIVm8,51927
alpa/pipeline_parallel/cross_mesh_resharding.py,sha256=5D1uIIlC6MjpNdGMWdtOlXX17E6MVPpRkHd1HAgV1Bo,64931
alpa/pipeline_parallel/layer_construction.py,sha256=95N3SzXCcvlNMj3wpKVttHXhouazko3KkUZ_qIUjcIY,25254
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=PR7rgPdgzejKRm8HhLRc06o60xUGHcSVgPIYH2Vg7no,22733
alpa/pipeline_parallel/primitive_def.py,sha256=xBHxemX2495TXaojkGHO7U6Wkadb5p2ze0VW5xBmfIE,7002
alpa/pipeline_parallel/resharding_tensor.py,sha256=WtlvoSCGyDfAwwDyhEgB_5Pk_to6P1ZAhjVYl-gRCu8,9179
alpa/pipeline_parallel/runtime_emitter.py,sha256=n37m-Sk-ig0M0Bwe5iCdkjoOB9gmtIrHozoNNgd8-90,46836
alpa/pipeline_parallel/schedules.py,sha256=eanbMIu-dpuMpYbSWnql9EQYxaTuFxTCFgDOs8O7uEE,15442
alpa/pipeline_parallel/stage_construction.py,sha256=GR0wPOdW5uMJSsZ7Xp2vUEWFDfnS2-DjmaZtEpAGgt0,34693
alpa/pipeline_parallel/stage_profiling.py,sha256=N7atDy8rh6XOE2sYR3ehDNCUQK7QCQ8KIEJ7XK8-4IU,36514
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-37m-x86_64-linux-gnu.so,sha256=8gnFK758oiu3g3Up_V76t67CvGRKCRCxxjjQZqDNCBQ,1300224
alpa/shard_parallel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
alpa/shard_parallel/auto_sharding.py,sha256=XCyugH1ntZ74JnhB4ekUr7Uix4ufj8hIHATnt7XC7PU,33249
alpa/shard_parallel/compile_executable.py,sha256=fLAK0x4q6HJbfqRcsq_bGcfIOhwyCkPB7deENWH8VrI,16238
alpa/torch/__init__.py,sha256=XnFF-2ijX0vDAYunV6fzZR9NKq1FwMKYK-UHq0R4zvw,5137
alpa/torch/tensor_utils.py,sha256=0KuwjXL_UJIYB0mb001qVmy2S5IE4GkugNhKwzKr7_I,3809
alpa/torch/trainer.py,sha256=QywY_KC36onIOf7kBCJPSTNaWLSifkgDnVzZBT0DH7I,4486
alpa/torch/nn/__init__.py,sha256=Zx8czV3vw0UiaFHpAlUxPsevYt-yHRfbfehaWEWYyfU,19928
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=NjC3U7L4j3RCPu2emLHeoChAXYM_Pc9ucyyYfUiWviw,18137
alpa/torch/optim/__init__.py,sha256=Qh74GFhfDvZHSdc4XTV8oCwzK4auRqbEyXp-Ajf_1M8,41
alpa/torch/optim/adam.py,sha256=qXmvXIWAfiViz4h4tOt8RCKW_X2FCCDKsqmCEu_4Fow,1409
examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
examples/opt_serving/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
examples/opt_serving/generator.py,sha256=aQdiHlyAWOmzlzLFgCZPB2ARDfoknJl6v3SyIhk7o0w,29815
examples/opt_serving/interactive_hosted.py,sha256=h1t4z93I1EaiCExbwYGZvrtp-lOpsU0nEtcKICrCj5Y,9192
examples/opt_serving/setup.py,sha256=35P7NDFoW3kvXg5YsIZZdHmu3sb9KRFbHKAotCtgrxs,1273
examples/opt_serving/textgen_demo.py,sha256=sRsXfjFOC5if3Ci6bM0SGR2uNBg_Cj2415p3hiSY_2c,1035
examples/opt_serving/dataset/__init__.py,sha256=ucwLHV5VTMoa75-Gr5x89It-UFxrYdmgwWzdgmGx17E,387
examples/opt_serving/dataset/base_dataset.py,sha256=OhMmzvHrroQ5wynB4JsI1sI8R3hIIexdx87sW5HL964,6732
examples/opt_serving/dataset/base_wrapper_dataset.py,sha256=8gm_RnYUSOuTWloODqiETwrr_mV49u9WuKCDpeZxI74,1884
examples/opt_serving/dataset/data_utils.py,sha256=ByzyEU1CUDW16_mkkNItHkxOqRhM98PboNPOFBComc0,6680
examples/opt_serving/dataset/iterators.py,sha256=kJsSKAg7zlNDLg7Rdde2zLGQmxlRm1MGPJKGgkJdGSY,19494
examples/opt_serving/dataset/nested_dictionary_dataset.py,sha256=yCfSov4bVavwQWmgDXBULAY946RkH_dY-XCuFoL_mPc,3699
examples/opt_serving/dataset/numel_dataset.py,sha256=O-FTuTM6SEy6zj6Fidt_K0Z7kRU6-vhKVrONhGr5P0I,636
examples/opt_serving/dataset/pad_dataset.py,sha256=b5gtuql7vU7ij0WkeeI5Vx94YM6Hn13lA9hUAIHP6IA,633
examples/opt_serving/dataset/prepend_token_dataset.py,sha256=JmSCdPUVR_89sXa_boic0fBV_nfP8_BH81KQO6e_XEs,916
examples/opt_serving/dataset/strip_token_dataset.py,sha256=URitGeNdQtpVAy_s3ZVGD9O-4Vm7qLwMEH8SYEFRAHE,497
examples/opt_serving/dataset/token_block_dataset.py,sha256=J5jMu0WRQ6zt55isIj_ZubqbknMT97WCkPcOCyIz-C0,8527
examples/opt_serving/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
examples/opt_serving/model/opt_model.py,sha256=tNjvwzud2OR6RFiQWQBt5XjK2x45L11ZB3aeSPPxMQo,36025
examples/opt_serving/model/opt_utils.py,sha256=JEifrvQG0TxhWSk9xXTR8QTEiqMDLMKj-jjXFyK7slc,1567
examples/opt_serving/model/test_cache.py,sha256=wHNMqD9hzIR_tf05HMaarj-35weZGY4EOBaWv7Lfc-c,2497
examples/opt_serving/model/wrapper.py,sha256=NWfsn3pwFF9LjnR-Thik_PcKk97aDrCIn0X9Ydc8ZkE,14513
examples/opt_serving/service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
examples/opt_serving/service/constants.py,sha256=FBcIjl88yHF8mE-RlwDTwiGzNnHu8uV4YkWeNjMSkrA,226
examples/opt_serving/service/queue.py,sha256=aGCAWunyEp8Qhd92woo4OY-VDSUJA8QxoyabmlZyIag,3286
examples/opt_serving/service/responses.py,sha256=5xOAxhICXeGtFYsGFcDl1qrTfWfINijdcNPBLvTkkgE,1028
examples/opt_serving/service/utils.py,sha256=vTj9V3KkVorvdp3uiDQQANJ8UVZPkk-XLwKVAfcFHDY,1771
examples/opt_serving/service/workers.py,sha256=XUcmQuX_PmGWiVqZ_SnpFpxcEW_iSyCTLM3akMy3iiM,1437
tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
tests/run_all.py,sha256=A-usUu4pGmaQmmtMG_g9eW0DbU0OlGjXtVHFXNjOXYk,3582
tests/test_install.py,sha256=jaAZiEY9dbdJEw_Ve75SoW3T0weoBXf9Inh00YD2azw,3696
alpa-0.1.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
alpa-0.1.3.dist-info/METADATA,sha256=YsrTBPUtBvqVE0I1L9ZL5BSec-h9B76ybwOq0nq1XHo,4365
alpa-0.1.3.dist-info/WHEEL,sha256=a8-lE8CDY3fTc-2Q7EKZhl4GYCxX5tnur3ksQusxWu4,104
alpa-0.1.3.dist-info/top_level.txt,sha256=F_cpv9V5pLyHowZG35Vr-0iGUFHzE3enbhvmrQ5q97w,20
alpa-0.1.3.dist-info/RECORD,,
