Wheelodex » Projects » colossalai-nightly

colossalai-nightly

View on PyPI — Reverse Dependencies (0)

2024.2.19 colossalai_nightly-2024.2.19-py3-none-any.whl

Wheel Details

Project: colossalai-nightly
Version: 2024.2.19
Filename: colossalai_nightly-2024.2.19-py3-none-any.whl
Download: [link]
Size: 1482417
MD5: 79d8dd42ff8d1ae84ae23c30eba4f267
SHA256: bbf88e75fd345cba7b30b7095a1f6adb4d306d5303c6e0aea5b2d3685a3779ec
Uploaded: 2024-02-19 06:44:41 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt · entry_points.txt

METADATA

Metadata-Version: 2.1
Name: colossalai-nightly
Version: 2024.2.19
Summary: An integrated large-scale model training system with efficient parallelization techniques
Home-Page: https://www.colossalai.org
Project-Url: Forum, https://github.com/hpcaitech/ColossalAI/discussions
Project-Url: Bug Tracker, https://github.com/hpcaitech/ColossalAI/issues
Project-Url: Examples, https://github.com/hpcaitech/ColossalAI-Examples
Project-Url: Documentation, http://colossalai.readthedocs.io
Project-Url: Github, https://github.com/hpcaitech/ColossalAI
License: Apache Software License 2.0
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: System :: Distributed Computing
Requires-Python: >=3.6
Requires-Dist: numpy
Requires-Dist: tqdm
Requires-Dist: psutil
Requires-Dist: packaging
Requires-Dist: pre-commit
Requires-Dist: rich
Requires-Dist: click
Requires-Dist: fabric
Requires-Dist: contexttimer
Requires-Dist: ninja
Requires-Dist: torch (>=1.12)
Requires-Dist: safetensors
Requires-Dist: einops
Requires-Dist: pydantic
Requires-Dist: ray
Requires-Dist: sentencepiece
Requires-Dist: google
Requires-Dist: protobuf
Description-Content-Type: text/markdown
[Description omitted; length: 28722 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.42.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
colossalai/__init__.py sha256=QjNSRxbLHBccMD2SXLxvzbA8F6lgSDA92MBtgw1IUkg 597
colossalai/initialize.py sha256=E8gqqCMfGtff81z5uIzCftOBkMTNgL2oq6uXC03JEm4 6537
colossalai/_C/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/_analyzer/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/_analyzer/envs.py sha256=Py9o5L4fjBZ10sRCovjk1E6-U6QbSZVkUpiL9eH9lKM 114
colossalai/_analyzer/_subclasses/__init__.py sha256=58kOYom_B8tlH3ZTT_v3-MYHMW0AZKHXYejBGzB4qqc 165
colossalai/_analyzer/_subclasses/_meta_registration.py sha256=av_yI07KyJ4lbOggQ6begE3V9TwTMEM308Z_G1X4Dkg 20868
colossalai/_analyzer/_subclasses/_monkey_patch.py sha256=Ao7dUVa039KsS9t6oAO7P1OkykffVjOIB4h2wIAoeUU 2088
colossalai/_analyzer/_subclasses/flop_tensor.py sha256=jHCKsJLiFIc1xC1UIxRvt5jinuh83DxeutuTCO37_lU 18677
colossalai/_analyzer/_subclasses/meta_tensor.py sha256=YHahcsMrn8tJZKnvdPgn9H9RhT9SZuvmeccJUJ0fzus 7589
colossalai/_analyzer/fx/__init__.py sha256=x2tiC9cI0QIbDshE_L1DkT5s9SlswDKydAHRtZ9JRpU 129
colossalai/_analyzer/fx/codegen.py sha256=-FFXn6F40lKNALiryXHkbL4GkyLTRFAmzFbstA_HhJE 18984
colossalai/_analyzer/fx/graph_module.py sha256=vwidMJWyq6_4iqIWlHqx3qEJ9RaK33D04R6jW1yvht4 9947
colossalai/_analyzer/fx/node_util.py sha256=uzJVcN8AoL-LGQI2HtUc1oDdpAUm8VNj0YcedUNnsrQ 8482
colossalai/_analyzer/fx/symbolic_profile.py sha256=uyJBw8RTP1ZbtQkxCbfEfbCoetqOZ22ubYibI66SjYk 952
colossalai/_analyzer/fx/passes/__init__.py sha256=f3KyqEG3KmJjia7dEJ7HvLZvzWzhdhm0awnwllOzGiA 106
colossalai/_analyzer/fx/passes/graph_profile.py sha256=e-7RdviBk6qrbS4AzjKYpX9HdJfZVIfi-NRgDf99VsI 12708
colossalai/_analyzer/fx/passes/shape_prop.py sha256=wfXi0jtkJdcjuOWlef4sHaspANEcvpgDttXT05SvQD4 9939
colossalai/_analyzer/fx/tracer/__init__.py sha256=mELaN-iIf42iQzH1MYsyUL4cNOhIM9pAdu8cUwf6YEA 63
colossalai/_analyzer/fx/tracer/bias_addition.py sha256=U-IDcRJC757ziYxYO5z0msjRbwM5QkjVmVe85ABJLy4 5415
colossalai/_analyzer/fx/tracer/custom_leaf_module.py sha256=AWPVHxNK_qxfo-UsTJ-SWKiLapKGbMHLsj9KquagomI 1108
colossalai/_analyzer/fx/tracer/proxy.py sha256=AEjiieA2D0fSLRB2rn8PkBLi4BEK6hoBftiktxRELz8 3568
colossalai/_analyzer/fx/tracer/symbolic_trace.py sha256=za27Fjsj6uOE8ZTT5UicyCeQgBYc6A4QvfIGjr01oaw 5862
colossalai/_analyzer/fx/tracer/tracer.py sha256=7TqZQrV2vVAgdm3y00xcs9mPMUsJy7HHUstXAcY-_sk 15702
colossalai/accelerator/__init__.py sha256=LrCASfi-Bxymt3SIdugLrqRC2a5oNvlNbRXuuvDKDFc 431
colossalai/accelerator/api.py sha256=AeMfISl1EvZzBQFvFhyZrI5ffYk1GZ6QF6JTK6ppryE 2148
colossalai/accelerator/base_accelerator.py sha256=tkPKbMbIK2kEcSP2PSG-__3DNPbpFscE5uospIb1fMk 9402
colossalai/accelerator/cpu_accelerator.py sha256=8Fz_HUqErkXIBYpfdI5mF7g_E83pEplskEBhp2P23fA 10154
colossalai/accelerator/cuda_accelerator.py sha256=itPfujzWCWVGV2aREooc4ob3UOTrcOoTNP4Jrjj5T5o 9442
colossalai/accelerator/npu_accelerator.py sha256=93hFbFdowie0np13XdIruEY4VZT987KNdak0SG5uxrQ 9453
colossalai/amp/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/amp/naive_amp/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/amp/naive_amp/mixed_precision_optimizer.py sha256=kKcJ13uK4GirN7OPlBqgNy_Uk8pCbOc4bt2a6LfC3Hg 7959
colossalai/amp/naive_amp/grad_scaler/__init__.py sha256=PkUp8oRgroDxGLgSdNx9mwU2FWyI0CDEbrI5cmt1kTg 222
colossalai/amp/naive_amp/grad_scaler/base_grad_scaler.py sha256=tslSs-brpl2yZr2imfE8kRhGwtrgTNA4RBNCdX-BQJY 2119
colossalai/amp/naive_amp/grad_scaler/constant_grad_scaler.py sha256=jdn1Rb3x3b_G0YGLNy8fCrWQWb4MJQXLXd4HQlRvzIo 735
colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py sha256=9Yj-RQ-EgEdoH87xKZqflk--PSqzNYuMWsuWEldbgtw 4977
colossalai/amp/naive_amp/mixed_precision_mixin/__init__.py sha256=jDBujVCdk0ZSqLf-5-B5Lbn0dYp3Uw4xblivdGdHJ8M 226
colossalai/amp/naive_amp/mixed_precision_mixin/base.py sha256=vXERxiWS2jws4cdaUE70HCame8R7mWe8gVNs8Ii6Edc 2523
colossalai/amp/naive_amp/mixed_precision_mixin/bf16.py sha256=nIZx4RmwClCxRiAW3ip-2azBjQTVxX9U_9QkHLqQ3XY 504
colossalai/amp/naive_amp/mixed_precision_mixin/fp16.py sha256=F4krLrCxsi1SNKCNdU1xCdS1bVzI-zuK3xX8gDD-ysY 2695
colossalai/auto_parallel/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/auto_parallel/checkpoint/__init__.py sha256=fBAeQf8PeluI8Z-IVe5Jq-8y73Vzi5wrIsQplhKUMEY 155
colossalai/auto_parallel/checkpoint/build_c_ext.py sha256=rd9KeoKQIfRUTU39SjIPmlhtQCeqEsmsPwS1nddZAjI 377
colossalai/auto_parallel/checkpoint/ckpt_solver_base.py sha256=eamR2JeHH2Pd5cl2FqdnNbjgHTx81XZh7wDAit5g3Ss 7711
colossalai/auto_parallel/checkpoint/ckpt_solver_chen.py sha256=WbMxDnIofWsqbBoTmQoUM0BzQ3p0-ICxLRL8hAScw_o 3468
colossalai/auto_parallel/checkpoint/ckpt_solver_rotor.py sha256=GxoGNxVVBfUQJqa-RiK1AgRgFMWopu2epmweqaw806A 18538
colossalai/auto_parallel/checkpoint/operation.py sha256=9PE35Y_uGb9ruQgTE-eo-2aZ6-dwsQe04uJ-kGL7niY 4921
colossalai/auto_parallel/meta_profiler/__init__.py sha256=2N1rmXcMngVwJ_9j93iMPIV3ffa1uBn7d3P9WuPDjqw 95
colossalai/auto_parallel/meta_profiler/constants.py sha256=Cqo-DXoSeW5RsLQQ82xdSRo26UW_V8kdpyxdGlIH91o 296
colossalai/auto_parallel/meta_profiler/registry.py sha256=rSyMNQV2p5t5p220YMScwHV01UXFVuuVzSGbzS-vY7Q 761
colossalai/auto_parallel/meta_profiler/shard_metainfo.py sha256=zqSE9QcSj-6K5BIGXNrPq7l95qlAUXFca0K79pM_Zao 4748
colossalai/auto_parallel/meta_profiler/meta_registry/__init__.py sha256=9xRbp9fvW8W7Z1vuwu8Eu3c1VC3UL3Z6veYL8C13iPo 241
colossalai/auto_parallel/meta_profiler/meta_registry/activation.py sha256=hYVpwa0VBtz_iPWNtuFZEHdIB8t_xaBQhAQl7u5j4Jg 3940
colossalai/auto_parallel/meta_profiler/meta_registry/binary_elementwise_ops.py sha256=MffrR2osE7ciGPWcpTEm4_-_ZwQ_C2DqLxC4NAuLA-4 2804
colossalai/auto_parallel/meta_profiler/meta_registry/conv.py sha256=8QDmDpfJlqZg8abPlca0KG2HB6sNs2SesbYNFPX6rzM 7571
colossalai/auto_parallel/meta_profiler/meta_registry/embedding.py sha256=S6xu_bpJRrgeBplVrdHscjzawLX0pju9a3kxwpl-fF8 2512
colossalai/auto_parallel/meta_profiler/meta_registry/linear.py sha256=mHpXFfxG5mVLzoTl07MCx9exbstuCfr_5RGYvuH1GCE 24482
colossalai/auto_parallel/meta_profiler/meta_registry/non_spmd.py sha256=T7YYmOAU2l70ockNnV51XDf8mkmzoieZ-tVVBwaUwpM 1028
colossalai/auto_parallel/meta_profiler/meta_registry/norm.py sha256=oJc5aB9rCY3gNli3dLszOsewCGFE7gMh5BCRPCTVHaU 9318
colossalai/auto_parallel/meta_profiler/meta_registry/pooling.py sha256=s08mQYR5bBr2xJeoeJFULdfJ_2Xixa9PWWP7Kwqrclc 7392
colossalai/auto_parallel/meta_profiler/meta_registry/tensor.py sha256=VAYwvdEVqRh100ZyItFah6Nm5ku4STQ_zlEU9qe8Aos 3258
colossalai/auto_parallel/meta_profiler/meta_registry/where.py sha256=17gQT3A6gMFM0Fn76GNe1G2N1M6LWAtlNKZLrD5kXIM 2827
colossalai/auto_parallel/offload/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/auto_parallel/offload/amp_optimizer.py sha256=Axz0PS-4bo1keHdNb2mTA-LXwVhJ2rnNwJmEHNjLaAM 6770
colossalai/auto_parallel/offload/base_offload_module.py sha256=TtOUib9IeJJWjWnT683_C930zmsGPwjb82ptlyPqxZw 3545
colossalai/auto_parallel/offload/mem_optimize.py sha256=6dvC3Me552ajwGqYmuxVADkkuRQSJV1RnufeU555_6o 2072
colossalai/auto_parallel/offload/region.py sha256=sTyrhe2Z8-SycbHYRoIgFPzI2jl709Yn2mf0hfrL0gE 5140
colossalai/auto_parallel/offload/region_manager.py sha256=HRhU1H1xhfMv5YiUCISQid59rLwio3Rx5-d-57oFGkk 20031
colossalai/auto_parallel/offload/runtime.py sha256=PAtrDSGEnGfIEk92cJ381yPjPtZMFXWuIR1yKVgMs20 9909
colossalai/auto_parallel/offload/solver.py sha256=IquhnOSow7oJC17RRmLTgfaaYYhDavK6Py-Gi3_rR7Q 18503
colossalai/auto_parallel/offload/training_simulator.py sha256=xVdztzaSXeom1V4AUFQY706y_xKPN18H3WI_0kRS1nw 17919
colossalai/auto_parallel/offload/util.py sha256=768WzwjMD2ExPtjZ76zZxvDz3Q7GtH1kmZ-8rBqvVfI 2793
colossalai/auto_parallel/passes/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/auto_parallel/passes/comm_metainfo_pass.py sha256=8_U4oq8rgPBcYv56KH6Jew6mXn00kFM6_RIrdR1FdGc 5111
colossalai/auto_parallel/passes/constants.py sha256=EozrV0y-qCj9qMcLeaM3Q4zm1op4MA70_lisVn-Z-38 417
colossalai/auto_parallel/passes/meta_info_prop.py sha256=FAUWZ5pX19IhnyJS64o_A00fW2w0jlMmj2gKiY-S_Zk 6050
colossalai/auto_parallel/passes/runtime_apply_pass.py sha256=NNFKCCzL-VaoaPKk4U7do8r-J7SZvN7TJDZ0duMcvoY 11328
colossalai/auto_parallel/passes/runtime_preparation_pass.py sha256=w3kgGlIR8WmN8KvA6NuLpij74v29ZAYd6nq5lkYv4h8 22439
colossalai/auto_parallel/pipeline_shard/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/auto_parallel/tensor_shard/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/auto_parallel/tensor_shard/constants.py sha256=_qjD0eGOj6LSeZWpPpOGG7eZXFxaUqragrjc_uAgwZU 2805
colossalai/auto_parallel/tensor_shard/initialize.py sha256=7DvwFCQylNmmP7eTzipuwmZAUdxLFYdR0-N280_vBjA 16998
colossalai/auto_parallel/tensor_shard/options.py sha256=YXXk_wZx-Ey8WeJplvb6rejNaiNi0Jw7T3XS2BsLAho 1522
colossalai/auto_parallel/tensor_shard/sharding_strategy.py sha256=wEzE4BwaGkEXyimp6DKwiBgXq61ZEVTJjBf39KaM9F4 10812
colossalai/auto_parallel/tensor_shard/node_handler/__init__.py sha256=1NzBSthk8yyyyEs84jeSNdQ0ry5kdZtKZBkI_hbBW-c 2062
colossalai/auto_parallel/tensor_shard/node_handler/addmm_handler.py sha256=NGX_pYlExrFDo2KLwTKwmCYxcXpmGYiz8LRFqeEHmBk 3731
colossalai/auto_parallel/tensor_shard/node_handler/batch_norm_handler.py sha256=Ntd5PHCdgZyLdCHoeKLtYjpx9LVTvmAdTUjvSoSJCLQ 3070
colossalai/auto_parallel/tensor_shard/node_handler/binary_elementwise_handler.py sha256=ZrhkuYLBJYdIXrQQkonZXZaCAkH1TIwTVWO6gnW8wr8 4988
colossalai/auto_parallel/tensor_shard/node_handler/bmm_handler.py sha256=uHZcqHOFJkYonTOKTYp1m_YURyorAdfN5_84z10dWU4 4805
colossalai/auto_parallel/tensor_shard/node_handler/conv_handler.py sha256=4Tfn0umn_DZPzG837tdBv2vSIlJxsgH6raKByrUBIDs 5563
colossalai/auto_parallel/tensor_shard/node_handler/default_reshape_handler.py sha256=v4TtkQM-F_KLypdSKCR9S6FnwJdFbHfoATZhreOZE8A 2798
colossalai/auto_parallel/tensor_shard/node_handler/embedding_handler.py sha256=1xjFDxqPsVUr940rCfXV5iEPzPaZ6SDeLKk01NzoXPk 11414
colossalai/auto_parallel/tensor_shard/node_handler/getattr_handler.py sha256=y3FPYdUfaArB0Mgi34b90Y_1_gZvD8XUkjwEVyL58gQ 1316
colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py sha256=yamo1SmLttAMAnVKxU5KkLDJJZDvDSjYGQnNNUXmf58 1734
colossalai/auto_parallel/tensor_shard/node_handler/layer_norm_handler.py sha256=JhqYZhmJSIZNrRG0Ny4ucHgaNmiemaf6zPF-rn5bsYA 1901
colossalai/auto_parallel/tensor_shard/node_handler/linear_handler.py sha256=RCxSCw7MIKokNxuDIXGIH2rLhUIN5aSFNfSY2i2PK0Y 13535
colossalai/auto_parallel/tensor_shard/node_handler/matmul_handler.py sha256=s9_-QsV953fQZ6t6XlUWUzPBF52qO0jrLUftl0EDT6Y 20347
colossalai/auto_parallel/tensor_shard/node_handler/node_handler.py sha256=NPbt9Jlbi8DoGoQ2kZzB9OO4I6ZMmr9oMXxLwd4dnGA 16308
colossalai/auto_parallel/tensor_shard/node_handler/normal_pooling_handler.py sha256=55HqDE3c-g64IU0-WbBtNq1YArnXPSuWAxJ7zOIiEdE 1762
colossalai/auto_parallel/tensor_shard/node_handler/output_handler.py sha256=e09AuSmXdDOHV6dvkplWoqkr3vFvNMSOld3KL1xLbhg 2115
colossalai/auto_parallel/tensor_shard/node_handler/permute_handler.py sha256=9oU5QHR_mnzgYRAZsVVjoh8QsD3a9ckOF39QQpy99AQ 2997
colossalai/auto_parallel/tensor_shard/node_handler/placeholder_handler.py sha256=aG_ne2GOUj9EIzEztmE_BNKmHOsHa0DEee9z-IaijuI 1466
colossalai/auto_parallel/tensor_shard/node_handler/registry.py sha256=AdJvjM-y1KR9e9FB6-EbxrrxkfhgV8oMdDLYvpEpeFs 744
colossalai/auto_parallel/tensor_shard/node_handler/softmax_handler.py sha256=X_vPu96NYlqk25b354VmyQ4CLHyquhYlKPbQecdbsic 1980
colossalai/auto_parallel/tensor_shard/node_handler/split_handler.py sha256=k4KWhpr1S7z9uNkZzY3-sDA169Xj1xtHtaBHi7cw2qA 2238
colossalai/auto_parallel/tensor_shard/node_handler/sum_handler.py sha256=vghLUZ0NW4Op3iRz7xhctVJJch17yeQ0aml_sANGPHo 3053
colossalai/auto_parallel/tensor_shard/node_handler/tensor_constructor_handler.py sha256=li28abfevOz__7TiQOMQ5_ikoHzpkuijSeFED4nj_1Q 1156
colossalai/auto_parallel/tensor_shard/node_handler/transpose_handler.py sha256=eotWXLMXSYFDTBV_M367MOXj0hrTJXXRz9uxthzZxoQ 2330
colossalai/auto_parallel/tensor_shard/node_handler/unary_elementwise_handler.py sha256=S0aH33N7RZFuQYoluA52J8umYDuoUdABhI_BNmBaCDE 1774
colossalai/auto_parallel/tensor_shard/node_handler/view_handler.py sha256=RGYQiHcCZB3y6kSrBpyI7lLVwOrqqhj0lp2321q7YGY 1964
colossalai/auto_parallel/tensor_shard/node_handler/where_handler.py sha256=9BuRwimyZKuxrgRxoACVBNXT9-BkmgzuFv5lMavhaFE 3553
colossalai/auto_parallel/tensor_shard/node_handler/strategy/__init__.py sha256=jtcbat8K1i9LFhjB-VslVp0jK46VoP_f4zV4QqgLmuU 2100
colossalai/auto_parallel/tensor_shard/node_handler/strategy/batch_norm_generator.py sha256=6Qa1cqAjcohM1SUDUN8vkFhzOak-BRYNCCRz8tdub60 14274
colossalai/auto_parallel/tensor_shard/node_handler/strategy/binary_elementwise_generator.py sha256=N8ioLr48GR8UPUuBFq45YtCkXilvlzbxZCNphFUbqEM 5162
colossalai/auto_parallel/tensor_shard/node_handler/strategy/conv_strategy_generator.py sha256=QLbAtK-D_xLpeHdE4SR29CCwUKkqjghPK1uiOrQXTZM 24078
colossalai/auto_parallel/tensor_shard/node_handler/strategy/embedding_generator.py sha256=TutcaIPm-sSJYSAfvEGinTZu-Shs5-nSAh8EgHmQzcE 12070
colossalai/auto_parallel/tensor_shard/node_handler/strategy/getattr_generator.py sha256=BE05AbFQ-MOI6a6Z8Q4_kx9z7bI_GiXy7DMjSU9y8XM 3615
colossalai/auto_parallel/tensor_shard/node_handler/strategy/getitem_generator.py sha256=cZCrBSz0dfFG4mqAN9Gm3q_oLgmPINRfYLNI7QuyrS4 7361
colossalai/auto_parallel/tensor_shard/node_handler/strategy/layer_norm_generator.py sha256=_pSLJOJ5mTiWSPiTFNQ2jDPAps-vWHsKxI5lWGaDZxY 9225
colossalai/auto_parallel/tensor_shard/node_handler/strategy/matmul_strategy_generator.py sha256=nrNGVEssHn8Tn2FYXvJbBSoeqAIdDSTOycTnXXpgQ8w 41664
colossalai/auto_parallel/tensor_shard/node_handler/strategy/normal_pooling_generator.py sha256=5THDUR3pJ2PbijHPSphT59vWuAQ8AQwr9A7s_H5jTtU 5538
colossalai/auto_parallel/tensor_shard/node_handler/strategy/output_generator.py sha256=XXqpvlnOlXoLI1Q7phvfm5xygqSyBfJreAaw3rmJJ7A 4597
colossalai/auto_parallel/tensor_shard/node_handler/strategy/placeholder_generator.py sha256=xF9d9N9hExW8ZUjHQsdvaCiQkzd-lqZY2VW_rYDWK3Y 3636
colossalai/auto_parallel/tensor_shard/node_handler/strategy/reshape_generator.py sha256=hZdSVgGof79uBdAleQHydTMqSO1tIsOGn-2rNz9msvk 18860
colossalai/auto_parallel/tensor_shard/node_handler/strategy/softmax_generator.py sha256=1L1yNDhaDDkjEGxLHA_2TxWGcwclCy9VugNemdYyd3o 4702
colossalai/auto_parallel/tensor_shard/node_handler/strategy/strategy_generator.py sha256=QWdpJp-m-6vM6WBoV8aSBn7zcYbMb1iZjszG28jsWqI 13053
colossalai/auto_parallel/tensor_shard/node_handler/strategy/sum_generator.py sha256=Kjv95F6VB33Xvyot0XnMcF8O1neirlaZgC7l7m9PR1g 4965
colossalai/auto_parallel/tensor_shard/node_handler/strategy/tensor_constructor_generator.py sha256=7ZGJjYv34lHb8ZqNvcwRVlIsAZfQMtx-RjhMsNybwzI 2278
colossalai/auto_parallel/tensor_shard/node_handler/strategy/unary_elementwise_generator.py sha256=33ZO1NskaGXzBMA6d8RDbZ_G6tdBY5ReFpkI7-vJ09w 3981
colossalai/auto_parallel/tensor_shard/node_handler/strategy/where_generator.py sha256=ejk7kKT-Y7hn77Rt1qrf80JR5DFQp0nEvn0YbRZy6UI 4158
colossalai/auto_parallel/tensor_shard/solver/__init__.py sha256=HUxcNmHE1NyFFxh8tJ_G4IQIYSv4xpoqKk-X0sVTNpg 238
colossalai/auto_parallel/tensor_shard/solver/cost_graph.py sha256=17NpJOm61m3EQ-DPh6kh6YxG2acqRcsRn5sfTptRiko 9987
colossalai/auto_parallel/tensor_shard/solver/graph_analysis.py sha256=ai3zmi2RpWYi_PSAi7vypQWn2jyiu99guIymA3yojbI 5767
colossalai/auto_parallel/tensor_shard/solver/solver.py sha256=xdWmpGnWMGQCUU-os2kHAbdTAw4DKcJ9sJdqshkLKb4 20206
colossalai/auto_parallel/tensor_shard/solver/strategies_constructor.py sha256=DIRyrwRnmndQjFZOomyxNxUjBwntxrzUNMyfq8WjFO8 8474
colossalai/auto_parallel/tensor_shard/utils/__init__.py sha256=yZtmphpOyNpWF4XMemNJpwW1g_HPaoWyDigudHoRS8s 1210
colossalai/auto_parallel/tensor_shard/utils/broadcast.py sha256=0TGQmnalcyp1lAEZB3DinleYZVsVPm-54h1J1I5pQhM 5899
colossalai/auto_parallel/tensor_shard/utils/factory.py sha256=jCyjhbePZ90amqNtyUQBrcyuiKi3vkX_HJl5nfxwlG0 8346
colossalai/auto_parallel/tensor_shard/utils/misc.py sha256=9T_0hq-J7PyuTvgY-etDBh4rtzVoHbp7OwgvXFJlSU0 3841
colossalai/auto_parallel/tensor_shard/utils/reshape.py sha256=e-UgRsc4Y5SIfS19kl7rQ1Ctfmjm4-6MxdmrVwBZBxw 9066
colossalai/auto_parallel/tensor_shard/utils/sharding.py sha256=-pQRigwOfJeNXOnUicBDIf3-gqu2fIeHxTk1myaU-kA 4408
colossalai/booster/__init__.py sha256=MfxfpmSR_-2fBKjkOr46UoJMDYngotCu2mZYiH-sSzU 93
colossalai/booster/accelerator.py sha256=8Qt7QyrnPS3f0Z30Kc067_Mc57C0iPX1YNmI4ug7EU0 1481
colossalai/booster/booster.py sha256=9_fhKYzjmKAEUpPtWv-Ueg0xuOflb024v4l9Z-rJRqw 16440
colossalai/booster/mixed_precision/__init__.py sha256=DMimOwQYzGM2lr0LuQQeKTfyBIuWJoqEMgJ55JS4tNs 1298
colossalai/booster/mixed_precision/bf16.py sha256=6rw-sp2hZZ1JBndxLSLoeeWZZjNHk3ua85Fy9O_seqU 102
colossalai/booster/mixed_precision/fp16_apex.py sha256=jBroUSBMtLojVdBLN9kCaBeyWxHFJHSMN4Mtx4QyKwU 3218
colossalai/booster/mixed_precision/fp16_naive.py sha256=Uh5vVb6LmLPORy54kG_xZh6r2YrYeBlk820DZ-mb8Tk 870
colossalai/booster/mixed_precision/fp16_torch.py sha256=l0j5ZmR65wNjx5QhR_lMdi77Ad5FSxdIB8TefOvtGV8 4824
colossalai/booster/mixed_precision/fp8.py sha256=Le7sO2IuaZb2WlEykeEX72RWYeYjmZ4ZUczZWRUUq18 101
colossalai/booster/mixed_precision/mixed_precision_base.py sha256=4PYHq5ZaeT0Uh9z4SvF5HgF_LBLRuDa9Iql4a6zGk4A 565
colossalai/booster/plugin/__init__.py sha256=lfv_hRRqiqjcKYk_gGa1ZkwBeZBzSuytsD_Fkmy83wE 529
colossalai/booster/plugin/dp_plugin_base.py sha256=M2q_BRObx10uQV4z7WndWgBsQsubY-WxvjfedvhMWCk 3098
colossalai/booster/plugin/gemini_plugin.py sha256=d8l8k5PQ-cD8e3pJW13ZNWJL96t_qOrIvfCXmL7Ch0Y 27856
colossalai/booster/plugin/hybrid_parallel_plugin.py sha256=FFy8acqM-J-80jjhPVib1YzTypdkcx6GIctAdH6TzD8 58503
colossalai/booster/plugin/low_level_zero_plugin.py sha256=DAdAd6rCZ9otHlWBfl0fh9o0h2G2lTPtLJyYkVzi9Fw 14888
colossalai/booster/plugin/moe_hybrid_parallel_plugin.py sha256=eEkYfMckrUhz8yPzyPz4v85sVEiEiFjGFkeNBdNf8vk 20670
colossalai/booster/plugin/plugin_base.py sha256=E1W336vhTl4tH7oR4Cu62wgv93GiUT6ewZe-r_kWRFo 2159
colossalai/booster/plugin/pp_plugin_base.py sha256=cYi0W9omf6Xb9rnPocxy-EcJf8HdGALzi-Oju5LrzaQ 559
colossalai/booster/plugin/torch_ddp_plugin.py sha256=LaXGKl3AdrNMKaGMvqnHKSfMUbpPUyqkKd5Hja_2e24 8120
colossalai/booster/plugin/torch_fsdp_plugin.py sha256=xldTgQmFof8JBmnzvqIpkbLU_ht2DA-gKOkOPECBr7Y 8978
colossalai/checkpoint_io/__init__.py sha256=gB5WVMk9Z5NPAwey_jUD88s5-nVgMhvkNrACUSlWhNE 318
colossalai/checkpoint_io/checkpoint_io_base.py sha256=Ovcec_IsE7ccbike9LD7JNzQ5up4zKRnkFBzf_nuHVs 14992
colossalai/checkpoint_io/general_checkpoint_io.py sha256=FN-Ol3ciUKT9m1RTFuyHQp-2cpPfgtMWZ2WAWZSiZDk 8613
colossalai/checkpoint_io/hybrid_parallel_checkpoint_io.py sha256=wRYn82T0aahAsUgogHWu0XWA7Xu90I5qwSoHIsJxIm8 42847
colossalai/checkpoint_io/index_file.py sha256=ankZbA98-PrMpfyn7CU6d1RAH6CtjDpdGf0WnoB3gsg 5758
colossalai/checkpoint_io/utils.py sha256=USullzk7OqIxlC0XmBacVKLB4wOVfQXll5GqS6EZ-Ec 29355
colossalai/cli/__init__.py sha256=9ByBOIdM8DebChjNz-RH2atdz4vWe8somlwNEsbhwh4 40
colossalai/cli/cli.py sha256=sCItG_RpxiI1NUFjWlLKqRTrP6rw-D7CUk7nozxM6tM 310
colossalai/cli/check/__init__.py sha256=O0XBw51qHKS9_wA0__AgWVBumCq1JI1UT9Cztffk4YU 396
colossalai/cli/check/check_installation.py sha256=KlBprHNR3xwZQGMqh9XxwDX4c3EZyydGQhjvw5keN18 8453
colossalai/cli/launcher/__init__.py sha256=246JBNn3Ko5LvawDAb9ikSFBHWlvxsMX8TV6Z3i29qA 3654
colossalai/cli/launcher/hostinfo.py sha256=ARY69bGsxdMxYHehu0Xy77uWXTUiObOXivbOKy_MHMw 3214
colossalai/cli/launcher/multinode_runner.py sha256=qrqg90noH9h3A6sNQOXki_FQdFLADjCyCS52YuXkvGM 4286
colossalai/cli/launcher/run.py sha256=P1q1JQ_bL9_PvwD4EZ6onHAlKJaz4VI7j2cNZj1AjlY 10530
colossalai/cluster/__init__.py sha256=8CvnM4zjknIHGUFJuCTCSsyPaEMjOLO5Biytr4iCzqw 296
colossalai/cluster/device_mesh_manager.py sha256=nRnhCv0XKlsPorKa37-nCgy9z9EMs8Jk_euWBwfzaOg 4241
colossalai/cluster/dist_coordinator.py sha256=SAM2cZGwsfrJkvJVgI8mT7JgFYJOWrqb2BSBvQqN4UE 7233
colossalai/cluster/process_group_manager.py sha256=TETGkVp5eK2D9Af8iXWT8a0kC6ZKHHT3Pg5BcOa8Zo8 2356
colossalai/cluster/process_group_mesh.py sha256=f5Xv7uDxmoqIdL-25sv7D9uw_51mLJSLuAVt2Y2xhfI 9777
colossalai/context/__init__.py sha256=YW9mR07x7A25sxYZrJdxjpyN-MMAa1uNYg95R5nMKEY 96
colossalai/context/config.py sha256=m_JTfrtgblrBoXzz4AQfXa1-XV2L7pAptdnMMiCwxKM 3153
colossalai/context/singleton_meta.py sha256=s41LH49G7NTZmvZtGJqImpB7NZ7eI-sSgoRYGhGsFhg 801
colossalai/device/__init__.py sha256=Syi4Pt-0VxmcqlpdalvgNi2buv5sOwMadLUbQDywAhU 139
colossalai/device/alpha_beta_profiler.py sha256=s5wLHHEkdS1WCLoOuyQ1H8L07MnSirhU0H2mY1lR-qs 17443
colossalai/device/calc_pipeline_strategy.py sha256=n_1EziRpBm3xfIoVx17etz9Srh5mWFOiiWx3pMz-Wos 5634
colossalai/device/device_mesh.py sha256=7gy-DXgZcJblc87rhTG-h5x01rUbfttKr-PK_lzaSEU 23616
colossalai/fx/__init__.py sha256=pJ024aIZ_eTlHcDE8UrLQzPwUejvjSIHM0fMNHHldIo 217
colossalai/fx/_compatibility.py sha256=tVZWK-qxHmZY7rTfq1lx1RtUUlxClWVB1tsXEOepppE 1548
colossalai/fx/_meta_regist_12.py sha256=vQ4QS434d02RHLNNd_E8_cfADADIxMfFPIzpdC7xxIw 19328
colossalai/fx/_meta_regist_13.py sha256=7MiFxOOeDzP2pAXquDzYnoY2Tkz2Ou6glXVFFYTxAtI 1906
colossalai/fx/graph_module.py sha256=N-7Ys55ZOaefMUKWbwL7Fi4JsiJc3itbE1k4TD8qunk 7438
colossalai/fx/proxy.py sha256=1JPBLRLdFp6iFWsOt9pGyoGxmMDjJgh96buSK_fOQzQ 4010
colossalai/fx/codegen/__init__.py sha256=tjtByUefja6Us_GWYNpzyx2qyCw2BgBFOLf6HaHV4qw 45
colossalai/fx/codegen/activation_checkpoint_codegen.py sha256=JU3OvBiz8mFNoY1fIsqhTT7OyY7EMpN6TaZK0yrYTDk 45217
colossalai/fx/passes/__init__.py sha256=nW7WvI_5vqbbfIuxTJKm-7h6SEsRmga2wMBon7fsP1U 266
colossalai/fx/passes/adding_split_node_pass.py sha256=G0NTWkzD129e9w6fHeVXdN5oJVbMBpsmXOJlShc76Rg 13496
colossalai/fx/passes/concrete_info_prop.py sha256=FFIXOkXumynGvTEL4-vsZGghWha1o2LyySbOBb5brCo 12261
colossalai/fx/passes/meta_info_prop.py sha256=mo_OGQXO3RW65ErjtdfKmzt7dztjs_hqA9di_zu4T3k 14066
colossalai/fx/passes/passes_for_gpt2_test.py sha256=tuOtbtfhYocaWUHP4X39LrbuFOLsuI0x9kVeYepv_Ew 15886
colossalai/fx/passes/shard_1d_pass.py sha256=ZxXDhDtdr_fMYd6i3vh3zZp8Fxw8G8rkqa9FUftg6D4 6888
colossalai/fx/passes/split_module.py sha256=KcSplqmE7sF53CtDtXX9pIaBBxJ_HB2AgSkK9RlFS3s 13714
colossalai/fx/passes/utils.py sha256=IpPWYrE1GW-Gak_YLCFv8uSOm5cbb1Cl-Aac1zhwJ2U 6169
colossalai/fx/profiler/__init__.py sha256=zCUjF2QswzaNaBikepbN7WB98NRZnAzCGLNQIsXURGg 768
colossalai/fx/profiler/constants.py sha256=4tvmgLL394jwbcutUyXni1ODhnHTcXMyaP_GDXSss3Y 871
colossalai/fx/profiler/dataflow.py sha256=FiEM_vkfKRzd65HLvHt_Vas72hya0LiQ7aYdRWHn2e8 6285
colossalai/fx/profiler/memory_utils.py sha256=wIHYRNhx3VPDUDNUiLHQJdD_qNdfQvD9j0IhT39EDiU 2232
colossalai/fx/profiler/opcount.py sha256=6zFKetR84Bx4RJlj4hdOwdkWbU9pU66ztiqoxCua1qY 13214
colossalai/fx/profiler/profiler.py sha256=DknyUsNcEbc5-iFIhgHifhQaEkuvrr2seslnrLo4NY4 15377
colossalai/fx/profiler/shard_utils.py sha256=ZVX7h8ggVaqG7v490Xg8orKgpFYRh4U8XQtRH1Rs5AY 4235
colossalai/fx/profiler/tensor.py sha256=hlWnnvXFQsJvE6zjSM_ZminBwYW7jyPMpjwL9JDyRnY 4990
colossalai/fx/profiler/experimental/__init__.py sha256=KZO17bDhFcSgcy7Tld92ghDU2Zb1Nxf0-O8Kk104NKY 282
colossalai/fx/profiler/experimental/constants.py sha256=VxsaBGgjc708pVnwbVeIeBLY_KSj5hj3-98abExDSSs 782
colossalai/fx/profiler/experimental/profiler.py sha256=pIjPkO6UyQMjRNH2IYlRbZr356CvteGlKkjdE_U7hUg 7062
colossalai/fx/profiler/experimental/registry.py sha256=DPJ6-ZLwJTa7aNHa9s45VwEmdTE140sNmmmRV2yqTiM 603
colossalai/fx/profiler/experimental/shard_utils.py sha256=DD_yTB-2ErE10jsfaJS4c_74O3BguCjBj0gHDbbKUpU 1050
colossalai/fx/profiler/experimental/profiler_function/__init__.py sha256=YP1wiJzuRbdOADlKTBikyp9m6MWv8ea52cKevhPct0k 211
colossalai/fx/profiler/experimental/profiler_function/activation_function.py sha256=9AJ2l37uH5CJBpru1tY-_eETmlBFGS3OVPHb-Zh1hWg 1304
colossalai/fx/profiler/experimental/profiler_function/arithmetic.py sha256=6mOM0l0zkVwr6BCS8sP0Q7Lr3_GNvzVp24H264DnnWo 3387
colossalai/fx/profiler/experimental/profiler_function/embedding.py sha256=_zt8bz9p3yVzFEu_MLl6oi4-_Ff35Vp_kNLiHKiceag 632
colossalai/fx/profiler/experimental/profiler_function/linear.py sha256=oRz7l2KIzg9HtLRvIL1L-hWRszmv7K6H3ZzU4a0NB3s 438
colossalai/fx/profiler/experimental/profiler_function/normalization.py sha256=7sSvWXEzUZBm6J8ei1Vh6ST1cd-asbLjqnR5Ok-z47g 2027
colossalai/fx/profiler/experimental/profiler_function/pooling.py sha256=ljfsB-_2RDOVvHNRnZHSATr5-yTVCnooPdfT6jcPM_4 1188
colossalai/fx/profiler/experimental/profiler_function/python_ops.py sha256=pWo__VLsGHv0STk9vQZV-S5ODhhX2VqccMfCQ4cFZWc 402
colossalai/fx/profiler/experimental/profiler_function/torch_ops.py sha256=mVLG4Qt2qLjss1ITcu77syk5y3HxU5YVZHc5h4FeQyA 2188
colossalai/fx/profiler/experimental/profiler_module/__init__.py sha256=6ZkSAdxBJ2oTyb9xStAiH0F0hbejSdS6749UE1_VmgU 252
colossalai/fx/profiler/experimental/profiler_module/activation_function.py sha256=P0bXieeLjb5SJObQwqxC6JeD-BgQF0f_2yL3QXdPOV4 1055
colossalai/fx/profiler/experimental/profiler_module/attention.py sha256=QUgY8RXCPvm5s0XXYaLO7nu73ef9TCjx9hRGVrGNmnY 2037
colossalai/fx/profiler/experimental/profiler_module/convolution.py sha256=pQ1iCfsgUmMmDhQMoygZ2xQohY00BrTyZ7zKWt1raOU 6708
colossalai/fx/profiler/experimental/profiler_module/dropout.py sha256=I302m1JiYCu8yY1F4t2myu_GhGerIiXyfHCekhqZoD0 424
colossalai/fx/profiler/experimental/profiler_module/embedding.py sha256=UuZ2V8J18PB2LFV7qGHq1lP7N2TP0xaTT96MnJOQLDw 431
colossalai/fx/profiler/experimental/profiler_module/linear.py sha256=PjHNHA-anVvOTQ91wPAvyCZ18ct3JVUjm5db_Vgb2GM 495
colossalai/fx/profiler/experimental/profiler_module/normalization.py sha256=jdENfuWwkqT4axna_wAn0c5BAc7bqtTLKtQkCIk8YQs 1582
colossalai/fx/profiler/experimental/profiler_module/pooling.py sha256=6QvOViCIhcf9WUfBCJA4bB06Rh1drvccGcm2Zu6nHdc 1013
colossalai/fx/profiler/experimental/profiler_module/rnn.py sha256=H6j-YzHaiKo6qN7MNjDsADA09PeGzF8jZe8anEGjhic 3020
colossalai/fx/profiler/experimental/profiler_module/torch_op.py sha256=Devouy9tEPbRtxtaUV3bYYnnKk-NsNn9ozTOsmsxLw8 271
colossalai/fx/tracer/__init__.py sha256=bDmJyhRnG_dPHmVAxGq2YiK3NrcpmzzvgshBKDt6Sh8 201
colossalai/fx/tracer/_meta_trace.py sha256=xHrJHeKqyJqKa4-LcF7hop3qfTQwBLwvPKwjlJ9QLxk 4850
colossalai/fx/tracer/_symbolic_trace.py sha256=Kk4EkgqWCgRZkbkNDzxDLR4GOHIOLeSdfx17Khkr2lU 2197
colossalai/fx/tracer/_tracer_utils.py sha256=VJS18I89TkZdloptzSV4gkTUVuY9m8p2xpHj_2DnT0M 1479
colossalai/fx/tracer/experimental.py sha256=gm54MzR9Beh48CKZgHxDA4399Toes3mOWeJ2poT_Li0 26431
colossalai/fx/tracer/registry.py sha256=IRaUVo8tMqweieWBcufub0_J0D_6bxSIvdlRvfF7NYc 834
colossalai/fx/tracer/tracer.py sha256=DWmAcbG0YVRi9CrKAS6hNRyO0uaqfYfd7ZuzAKT5Mqk 24642
colossalai/fx/tracer/bias_addition_patch/__init__.py sha256=t5j8H_auhAXzen1IYCM8GtGyY0VRRGhLsex8Ef71LO0 90
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_function/__init__.py sha256=fZQG-ZL7dIw0chqzhfD-L9HpuBseFN2lb_Z8xm7pfAI 193
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_function/addbmm.py sha256=ZMBf2BemSrFj209nCLlqPwTimmYORVi4IQVHCVbMhXk 2798
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_function/addmm.py sha256=lMPOGKxIksj2b6E2vn3nJ02jxxUzjh0HZfn1fg0kGRo 2171
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_function/bias_addition_function.py sha256=ScJxFfG5RrEfMoCd6SBSygsFO96M2pzc74343pWI0gs 4445
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_function/linear.py sha256=9rhACUXH_dHy2uRjxUoxvelmnTzAxWl_xaddqbKXkoI 745
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_module/__init__.py sha256=YWu6jZb7pZdOBZg8U8S8pbtXbilZqgb-oaa2L_ii8JM 78
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_module/bias_addition_module.py sha256=P9n2FIxhRZCXijLNnGRO_IYC9q47FRsXt8sUCFlMT_U 4395
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_module/conv.py sha256=zKFLT3TSz0Yg6TjARO_KQb3ERKsdi5R7qj7xqE7-Vz8 2370
colossalai/fx/tracer/bias_addition_patch/patched_bias_addition_module/linear.py sha256=MItxtYbHWp73pUX5SII3WxLrv2r_mJ52y_CVPwSVeC0 503
colossalai/fx/tracer/meta_patch/__init__.py sha256=cMY7bGlj0Ocj4mWqQQ-6jg1ygC80gp4y5SZFlES_Yqk 62
colossalai/fx/tracer/meta_patch/patched_function/__init__.py sha256=yr4v9-RpCYkbBgA3GDQ50cLOOx7hxXwCjLnQ2Cy2dTI 167
colossalai/fx/tracer/meta_patch/patched_function/activation_function.py sha256=GiIJr5yaHECXt2KqksycX4xyryevnVYvTQzvDLZav5g 217
colossalai/fx/tracer/meta_patch/patched_function/arithmetic.py sha256=yt3k3AT6Ogh_tqZ-yE2_Qp0cJ8oLDBAB_Gvp4NYnsUs 3200
colossalai/fx/tracer/meta_patch/patched_function/convolution.py sha256=INMXmewSsgzpAD83748s0EL1vhXOF5g2b_3r72548DM 5707
colossalai/fx/tracer/meta_patch/patched_function/embedding.py sha256=e98igwDVNSi3l9sZCFXp2X66vWHsfUgKqpHkSwOo-Do 339
colossalai/fx/tracer/meta_patch/patched_function/normalization.py sha256=L-51IHPFeg_KMBHmH-j5lt0C3UNGsZ4cVrro7o4ym30 517
colossalai/fx/tracer/meta_patch/patched_function/python_ops.py sha256=xTNdGastkJ1mdUzLpZWexf3cCtM46sCwoktcjvyFLw4 2047
colossalai/fx/tracer/meta_patch/patched_function/torch_ops.py sha256=Ze2zWFmia62Y-UeVxs4kvCRmFmwdjiOE5-U8F4W9NBk 5622
colossalai/fx/tracer/meta_patch/patched_module/__init__.py sha256=hqqqXSYFL-F0Ll5c67aOTkMGg98OVFwEASTEdBPR2hI 180
colossalai/fx/tracer/meta_patch/patched_module/activation_function.py sha256=bLpf2T7WUD9_qsEZIo3IxibjIFA7V9z7K46A-PnNwmE 428
colossalai/fx/tracer/meta_patch/patched_module/convolution.py sha256=Y2L4UQge1fFlFwcYRMJ8BUEJXnIwWYVLUT213onjUuY 4752
colossalai/fx/tracer/meta_patch/patched_module/embedding.py sha256=wcz0XgU7oc5iOCGJl2fMkROBDWub01vIVAx3V19aBJM 254
colossalai/fx/tracer/meta_patch/patched_module/linear.py sha256=W-WE-B8w2f8eV73oQmADsXNqoUQo8Ie89DIX0fRBEhA 400
colossalai/fx/tracer/meta_patch/patched_module/normalization.py sha256=ZMaBL2rcda4T-2OY7P_ruvoOUyPzCeoFZgHp3JMR2YI 1129
colossalai/fx/tracer/meta_patch/patched_module/pooling.py sha256=d7Lr523paZm1v6UNSyZm9cfv5TknG9jg7LmB7tCYDpc 6769
colossalai/fx/tracer/meta_patch/patched_module/rnn.py sha256=KJAJ76JcOYGlj645Pxz2p8qdwBrcZnq6bTzWJ-a1vM0 644
colossalai/inference/__init__.py sha256=rU47A1GQ0AierhPKRwwPcjDrUL0mH71GvFby5kH4Kdg 235
colossalai/inference/engine/__init__.py sha256=lfxlfIOsZaQWOg649GTxzHgYqbRqWrdAoChdh8aL13A 67
colossalai/inference/engine/engine.py sha256=1HjOm6EideN9rK8lCYu7di4xPefo6kIuURVS4eOVluY 8315
colossalai/inference/engine/microbatch_manager.py sha256=UOb_6nVV87AaPHq1fFtTOXEb-Yi5QfrYngpWHvVZh1s 8923
colossalai/inference/engine/modeling/__init__.py sha256=bcNteuR8pdfgGh2Iq4xj6xcKetvvMXoCNjUp9BS2lyg 225
colossalai/inference/engine/modeling/_utils.py sha256=2MsD0iAGx8lFbdsbgvxBvMu4b8AANgdqCf2X1DJ_wvM 2706
colossalai/inference/engine/modeling/bloom.py sha256=viPoPS_Hj0iJptpL2GARDoka0w6nI2g4___8UFBt-Pk 19778
colossalai/inference/engine/modeling/chatglm2.py sha256=BQTqTfq4SxoEn4Qd_AEZZdHu12vAY4-HxFdFThkdarQ 20627
colossalai/inference/engine/modeling/llama.py sha256=66IxvUDoaoV1Hezl_0idClyRRL2rQ5zT4fhq7nZ9O9w 22054
colossalai/inference/engine/policies/__init__.py sha256=nxS2F3FRTCxEreeZqmhEH7BMgn61Hbr7h_FIIwQHWRw 360
colossalai/inference/engine/policies/bloom.py sha256=oV-TVD8Atft4Bqd_d7WxkQspLVZeMH-2EFFWnMSebyo 5437
colossalai/inference/engine/policies/chatglm2.py sha256=tq66HK-ul6DyrJbLboi_xXJyTCD1eqbWQENaq7aKXng 3467
colossalai/inference/engine/policies/llama.py sha256=zmIntkcbk76r234T9tbKrGWcSbgBDwa7BbBOfSHy5w8 8514
colossalai/inference/kv_cache/__init__.py sha256=P1tDNWEbrY9RMEemE8D_Ri9zQ5qFf3TW7Pqrc3enL7c 90
colossalai/inference/kv_cache/batch_infer_state.py sha256=Brchl_focDqhgUEfc2Ny7wBuQIbUQxNI3DDObnqETG0 4483
colossalai/inference/kv_cache/kvcache_manager.py sha256=lZhFjIsWonxcLjf1Z46P3sXrGPPa3rbY0fiAK4ib-bw 4582
colossalai/inference/quant/__init__.py sha256=ztahESMhIjAYXBTO8ITttxpAn2xiME6vP0iBUZN20O8 61
colossalai/inference/quant/gptq/__init__.py sha256=eQlul_qBrzaDgK5kZt4sif9PdrDhep1zQdHlVRlfNYA 155
colossalai/inference/quant/gptq/gptq_manager.py sha256=MRauISb7_LFhyxUlVQ7lFSgIWnzluSOVeS09tUqVYBI 2423
colossalai/inference/quant/gptq/cai_gptq/__init__.py sha256=YLdqHUDy3fVLcTsOnLJvpYLpWKpKAxsY6zLClHDMG7Y 372
colossalai/inference/quant/gptq/cai_gptq/cai_quant_linear.py sha256=U1yfMEK2wVo6oGPqTcLTlImdciPyS-2hiQQ4h7oy-Vk 15011
colossalai/inference/quant/gptq/cai_gptq/gptq_op.py sha256=prwdAwLHnEtmsSuQMUIyTvLQDeRcV4zygS4wqW7nUsc 1629
colossalai/inference/quant/smoothquant/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/inference/quant/smoothquant/models/__init__.py sha256=U37MH5PYzlC9VY4y4Oo1zER6fAnzd2-Edg9zpMhAcsc 297
colossalai/inference/quant/smoothquant/models/base_model.py sha256=6iCS-F47hAk7eO65b0KNWrSIxHfTAad0HOeV0EhtwqA 20063
colossalai/inference/quant/smoothquant/models/linear.py sha256=b828B8L0jdUjP2GQtSf4ZtuTdrSVqPOGCX1L76OGcu4 6500
colossalai/inference/quant/smoothquant/models/llama.py sha256=S_wDORBq0z_JVJL-CwFfy1HXLTTX2uFKJA7O5-2SfrM 35519
colossalai/inference/quant/smoothquant/models/parallel_linear.py sha256=7y0go8YE1BkNG9fYyj2ULhlqFq_gFP6-iHzlNtB3OH4 10276
colossalai/interface/__init__.py sha256=qtpeUA7JX1Z88CJ4r0Eep8eamOUEJukMxML3XSWJoSA 152
colossalai/interface/model.py sha256=ilEZCJ5RmEQaPeQSKtLILw_HjCJi8MZ7NawYzMD_fmU 851
colossalai/interface/optimizer.py sha256=1dDAWooOECoVhD0jp94_kJQt6lGHPes23cfddLyO6QY 4120
colossalai/interface/pretrained.py sha256=EBlJkuR7rUFbEj6AP6OosBKYyvzYl2_BNNBQtv1T09Y 333
colossalai/kernel/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/kernel/kernel_loader.py sha256=qIomHo6iuzNOKbX16Lfo4s6tSun3WQmlzYwkH76VS6U 3395
colossalai/kernel/extensions/__init__.py sha256=xqgUCwrCAPEeRPuOazz-bKqkOSyeFaaLXUDuqQvHESw 1156
colossalai/kernel/extensions/base_extension.py sha256=KsgyeKIh602OaCFnecPQMs_tCw8nFJ4fCIP2HSMAS6w 2517
colossalai/kernel/extensions/cpp_extension.py sha256=EF4kXMI-cuWx5uqtu9vNInxybwYkaIh48blcVamkrmU 4730
colossalai/kernel/extensions/cuda_extension.py sha256=ECloTWDHzpjymdLEDhnu0qOAAX0M9Ub0Nlgz06YWQpk 3896
colossalai/kernel/extensions/triton_extension.py sha256=D6zyGlobiqXiCzmxMwtOxHVu5OBXk7KSDnWYvz0TZKE 589
colossalai/kernel/extensions/utils.py sha256=qpThSvgjvHGHWcAM656bTZATpruvdXITsbftqoDKY0E 8283
colossalai/kernel/extensions/cpu_adam/__init__.py sha256=hUCxmVFMCigXdFoRuCUYEvFUtun_m4dR1A1AeAbC2yc 151
colossalai/kernel/extensions/cpu_adam/cpu_adam_arm.py sha256=rAAv2eaVjIT6Z_UYOmtyZQ62ayehqla90Tqdf554T3k 1043
colossalai/kernel/extensions/cpu_adam/cpu_adam_x86.py sha256=T9AlMnEU3bcfPA4NlJ7VgGTb_g7FtXvn192OHnNIRoo 1660
colossalai/kernel/extensions/csrc/__init__.py sha256=UzC58VkROMaJgM1Vs_Upx88Ta4QI_GJZYR_lpdMaySo 350
colossalai/kernel/extensions/csrc/scaled_softmax.py sha256=MUHkpayTNpJSPfLjtmSzOkMNQOUaN_t8YI1tWVymz88 6588
colossalai/kernel/extensions/flash_attention/__init__.py sha256=xjTAEGepotS8ezBSU_DDCcRlm7x0hKnJx9TEmVFuy5M 527
colossalai/kernel/extensions/flash_attention/flash_attention_dao_cuda.py sha256=VgXoWL7w0ReGds5WThnOf6T2YLkBRyPDy9FzGe_93CE 3502
colossalai/kernel/extensions/flash_attention/flash_attention_npu.py sha256=1QeLKo919POHDwhNLgKrn14o_toJ-QnyZwE0jCBmEmA 2360
colossalai/kernel/extensions/flash_attention/flash_attention_xformers_cuda.py sha256=w83ED0TebXXmTiDuJc0f6SlZR6MH8T4WRRAPX7ozbec 3622
colossalai/kernel/extensions/layernorm/__init__.py sha256=fUNamrXwENT5vcBLrPGPkjBcKA2rzu2SEDxO5Cyq0d0 88
colossalai/kernel/extensions/layernorm/layernorm_cuda.py sha256=iEWWZ-Q5avyCqy8KwZeEhXIh87tbJm2mVDzsNBRaDYg 822
colossalai/kernel/extensions/moe/__init__.py sha256=XEM9bhqREZWBCag8thn6nFdLPuQuwW0pxmcTjx9gRHg 70
colossalai/kernel/extensions/moe/moe_cuda.py sha256=OocQN4_roGwLJqSEdbrDqfKWG3NInYtSi9EyD_BvDFg 959
colossalai/kernel/extensions/optimizer/__init__.py sha256=H4kcmlwol0VR6dM1Vfq4U0qWR9ZMhQVX2x1K1WS2XLo 104
colossalai/kernel/extensions/optimizer/fused_optimizer_cuda.py sha256=x2lHMOkWsM1kXFUodIP3kjgieiSlzFkkmMisAA7_MXk 1104
colossalai/kernel/extensions/softmax/__init__.py sha256=sTv-jI2bukCzTczJC0EbAzN7Ag3oqqS0lYQnjQfAm6c 271
colossalai/kernel/extensions/softmax/scaled_masked_softmax_cuda.py sha256=A37wtyB_MwfvZcAfHeTXMrVlbPo93lxXOaqqa2CQN8U 1027
colossalai/kernel/extensions/softmax/scaled_upper_triangle_masked_softmax_cuda.py sha256=zrvg39Gu_TUssvv03bxJx5vAFoty2WOO_KQOlfrRkhY 1098
colossalai/kernel/jit/__init__.py sha256=UP1cy7gB1wa2EDQy-ieoYpZqDUeZB09qSGzIBqYEEMs 317
colossalai/kernel/jit/bias_dropout_add.py sha256=lOPNM_aBak7fDMN8-j4QAE3UIccwTkRKfi4B4Pdnl-s 670
colossalai/kernel/jit/bias_gelu.py sha256=PT2pHEWYpeWGrg1NuQzjOl7uaC3JMGrbUk119NAygcI 1358
colossalai/kernel/jit/option.py sha256=Cr8xXWbYoSbXUc3jdCUBPN5SNBEebNk0XC7AxTjp27c 3591
colossalai/kernel/triton/__init__.py sha256=a-Sg8SsGrfs_gDiKNMZFvmnYLHzcy8McpEC98WySlow 1096
colossalai/kernel/triton/context_attention.py sha256=xVZxOMIl0r2K0Au75_GcvCCuw7q70IJHArZLDdlo30M 14518
colossalai/kernel/triton/copy_kv_cache_dest.py sha256=saKM8W-CfE64Trjyuyxd71Bt686T3Abzs6qCF45NSz8 2349
colossalai/kernel/triton/custom_autotune.py sha256=13M3HlxI0yEFFHOl8QluQslbbvR5vWr_K5SpesPfL0k 7022
colossalai/kernel/triton/flash_decoding.py sha256=2f-zlXpRmwkhlQYkADJbiTq8qbAcgkUNrXy_4xJ1--k 2397
colossalai/kernel/triton/fused_layernorm.py sha256=YdzKdFzelWjNxhG5kR88u2pVLflceNhkacKbPP31guc 2960
colossalai/kernel/triton/gptq_triton.py sha256=9eSiMhtN0DGbKsovFrEVX4sXSkyJAjjlGe9wtBI22CI 18024
colossalai/kernel/triton/int8_rotary_embedding_kernel.py sha256=fRfDH8fXlkv-xUEhguaMD1yWOhR-yBvZdCfVFrUfjb8 3280
colossalai/kernel/triton/llama_act_combine_kernel.py sha256=pI7GqOwFh4nvaxEPCMmxwHcYb-EtKWIWeGf2w2udfB0 7353
colossalai/kernel/triton/qkv_matmul_kernel.py sha256=ttHJwzDjHJGCW4obqUaRLdf4g-xFiWR6qhWP1Bgl-R0 4967
colossalai/kernel/triton/self_attention_nofusion.py sha256=W5CiNxQHM4rmyaatF05cgqTL3evlPUJ_fYJ4H7X07J0 5947
colossalai/kernel/triton/smooth_attention.py sha256=xFJ4AYbU9E2MlH_XeOyRMi4-HsBjUy1a0qUcBEz0FI0 21773
colossalai/kernel/triton/softmax.py sha256=seCJ9gVEJXeU8bOT1ae8G9hNgORbAxj-zwUXFVfmgb0 3744
colossalai/kernel/triton/token_attention_kernel.py sha256=gs7mCj22xXPj9-tCfpTTNRG-rpcwKyXjB3tPO8awP94 7858
colossalai/lazy/__init__.py sha256=3bOoAKO_0A5BFVq03CQ40XlwdSHNKphvY5-QwuSw70I 107
colossalai/lazy/construction.py sha256=kpvBwdG5DZYb7R4ubOHFT-2NRMV4RADfv1202hMuzFY 2187
colossalai/lazy/lazy_init.py sha256=AlfTm-XpN47ddJHC1bMk9stLxV5rw-4RAkFQaRZ54YU 25011
colossalai/lazy/pretrained.py sha256=pQP2hTjT7kyMt8mWnGgkLNTZI002nuWLHEPGQz8tFEk 13895
colossalai/legacy/__init__.py sha256=HmAbti3Isa6qWqbToiz2MWkOlgaBSx0NlPRX8JhuRjo 301
colossalai/legacy/constants.py sha256=aya77IAdAKUgYZIW2LMfPvXWMfXyFsbwuwjciRd14WM 978
colossalai/legacy/core.py sha256=uJsreDMypAztEfLdYf5umqrYs4zArQLjq07O5hNje78 149
colossalai/legacy/global_variables.py sha256=pqwWrx31vT-W622CP3zqTAKjWvWlxUILwNHCtmW4TwM 1993
colossalai/legacy/initialize.py sha256=S7Q8Mr05Ij1q_shR1zHQIK4JOd1as23OI7AAqJ4So90 19842
colossalai/legacy/amp/__init__.py sha256=crG8P3RlzDDzpo8AQ-j74JrA2ndKSJrmjHRBWDo0dow 2310
colossalai/legacy/amp/amp_type.py sha256=sWZZmvsUy2tN9Ur5SnmA2fDhO0tuXqpf6YE_25RvWc8 153
colossalai/legacy/amp/apex_amp/__init__.py sha256=DjE0czvUg5Ve8Qn3YEAz8P58vjPbAC4JXf_4BubY8GA 1637
colossalai/legacy/amp/apex_amp/apex_amp.py sha256=Wymecw0kTh8ATp7aMQv0-fzFH4Wh1wkVFfr-PqgPYV0 1073
colossalai/legacy/amp/naive_amp/__init__.py sha256=GnhBL7EoKmkKA6ZGbFFrDYufDYpIez3x0RsUbOWpDXc 2453
colossalai/legacy/amp/naive_amp/_fp16_optimizer.py sha256=kjwiqttyTq08oqkh9k8Q-VVusru5K5SS6k8N-qq0CyQ 13449
colossalai/legacy/amp/naive_amp/_utils.py sha256=syDatprobf-LpIcQX1mcj86r01xawqO6HGCLv5oiJZs 1741
colossalai/legacy/amp/naive_amp/naive_amp.py sha256=6OcRVgUGwb3wL4Bfn5yTWtuBUcFXznP4yStg1vYtTIc 6081
colossalai/legacy/amp/torch_amp/__init__.py sha256=8OU-NebslKgU9I7Avy1wb_Tu6SkrQPi2kBEFxFZfeRY 1538
colossalai/legacy/amp/torch_amp/_grad_scaler.py sha256=L-XycGoof-MFHZLuHRjkPKKSvMdxpUHtDxz4hrVNXUU 26771
colossalai/legacy/amp/torch_amp/torch_amp.py sha256=ugSr89w4CmkSkWVxInQv3YLSixiLlz-aAOzlP1bNG2k 3368
colossalai/legacy/builder/__init__.py sha256=FyNjouMsGmXjY4iyIpOoSxHv6xyawjYsSp5sVOnIVtA 166
colossalai/legacy/builder/builder.py sha256=-3ssEHHc3R6h4Lp-IDX0RsdCd_cE42JFulNM5l_P-OE 3025
colossalai/legacy/communication/__init__.py sha256=wDGxqDq17o-5MhDj4lJlBM2dxvocFDUgGu_Cdux5HcA 868
colossalai/legacy/communication/collective.py sha256=_slMsGaPrxAdXKi9a1wOnqpcoQZjiS_dW4ukk3UmMJc 11387
colossalai/legacy/communication/p2p.py sha256=6B11Jb1R9ArSzyU1nZMpfzdt4ywweuMtWy4iZhoAGTY 17484
colossalai/legacy/communication/p2p_v2.py sha256=Km7Kkmh9HabX_SGxSFK-M1jIg8STJKH13VP9X1zqT6I 9047
colossalai/legacy/communication/ring.py sha256=idQtyETdwYGvjrhCSNjQE74SrSHTRVKGvlRA6OC92xw 1945
colossalai/legacy/communication/utils.py sha256=hqJ8kSXmprc6JTzmQuiAKuBBSkJCFsn9Fr0HEt2uqUw 5151
colossalai/legacy/context/__init__.py sha256=fOPwmQElFPbYtYLGLvU7tIsPRWP3AaG2MMqpp6YK5Jw 149
colossalai/legacy/context/parallel_context.py sha256=MCTbsthXWMjy19nE3HR9u0AA8pSQduFKFD7erSFbUuM 24229
colossalai/legacy/context/parallel_mode.py sha256=h-LSmsdR6hBNdlf5Sqgs47pXfLwXtLCYyz8s3g4zzXk 1142
colossalai/legacy/context/process_group_initializer/__init__.py sha256=OU9tUNb1HCurNyrU3dyIs3vWEhIE_MO5pij-ZHW603Y 763
colossalai/legacy/context/process_group_initializer/initializer_1d.py sha256=mWBi5Nvf1SXYEbUap9WT1X_fzM45NHUIZLevdk-oakY 2144
colossalai/legacy/context/process_group_initializer/initializer_2d.py sha256=QKdui3DYACIdhYNYOwYPJUlrgwe-UeWCbtU56wzuIIk 6269
colossalai/legacy/context/process_group_initializer/initializer_2p5d.py sha256=YN8afhm-CmnNILlk5gdNmJT9oKHSWwV6vMVkUx0d2tc 12944
colossalai/legacy/context/process_group_initializer/initializer_3d.py sha256=vgwFILu8G2N7LEO4LvgRRIsqfCCEATcuEmsD2Jx0MBQ 13290
colossalai/legacy/context/process_group_initializer/initializer_data.py sha256=Y4XRGkEgGBCuJr5xtKQ2Y6qGyPPv5cnbxrdiL-_rIb4 2041
colossalai/legacy/context/process_group_initializer/initializer_model.py sha256=X2naldOjGL1Gf2ZaINbHMePqgQCMwyIidC3_tDc67Fg 2174
colossalai/legacy/context/process_group_initializer/initializer_pipeline.py sha256=X2wTLhQ7a7uXCRMAG3tKOHDA9-QvoC5vZRb1pGp6Tag 2652
colossalai/legacy/context/process_group_initializer/initializer_sequence.py sha256=Wg6ZWNQKuQ97wn0pVKPA1_QXIYgJEDXpVhLr8j3_Mrc 4170
colossalai/legacy/context/process_group_initializer/initializer_tensor.py sha256=FzYK28BQ-hbIl_GoCEoNv9sxLttIuNlRWCZUAQPj7L4 2051
colossalai/legacy/context/process_group_initializer/process_group_initializer.py sha256=7HZnf7_-lxnwtMuFTdr7v1W4jpUi0AQ1LyxvXHl0WKU 1180
colossalai/legacy/context/random/__init__.py sha256=T_CMAv7JNi8HPOmelEfQw3aHJtGgxa3yOS2Wlo-Ibts 420
colossalai/legacy/context/random/_helper.py sha256=OV6ojQhomqj3kz6YwE6GHU5jTIVtwd3AE7vCbxJ7QNc 5204
colossalai/legacy/context/random/seed_manager.py sha256=lHyZY6V48iLVOAX9ZcpEgmP4-yD3BHnRaStO1SaASr4 3407
colossalai/legacy/engine/__init__.py sha256=xKxbzWKcHJ1dNFz3TgEZqjJbeSCYjBwirfnA6Q7WvFE 87
colossalai/legacy/engine/_base_engine.py sha256=jWLB4ef6uZIozoWx1bkgqeax3d5S9aZMTjVSDkom9QI 7784
colossalai/legacy/engine/gradient_accumulation/__init__.py sha256=NflrkomNSSy0xTh9HGyzGxH3LASZSNcvRcla6fpHHH4 2558
colossalai/legacy/engine/gradient_accumulation/_gradient_accumulation.py sha256=4yNTgrUydlB-f44LPsBwhQJpa04GU57E2iezHzekry0 10265
colossalai/legacy/engine/gradient_handler/__init__.py sha256=sxWucTVvCMhdl2Xjz7TNchjplQoo-KBI_5hoUGCNf3U 537
colossalai/legacy/engine/gradient_handler/_base_gradient_handler.py sha256=KV-T_pLk1PWCFfwRbS5DCi18jDH0HzdMnF43XDStQlY 750
colossalai/legacy/engine/gradient_handler/_data_parallel_gradient_handler.py sha256=3Mrm4DmL2nxHdzj8MKliGsb7Al3KGjtq1Ii5Lh-1yao 1149
colossalai/legacy/engine/gradient_handler/_moe_gradient_handler.py sha256=iO0L0epY6kz-AsBb-Zho1_8zEW1XthP2WJ1j1weorbc 2138
colossalai/legacy/engine/gradient_handler/_pipeline_parallel_gradient_handler.py sha256=QOKF-fyFh2u2uTofKGLVfhLNcPQ0n1ccWrqOYoKpIcg 2488
colossalai/legacy/engine/gradient_handler/_sequence_parallel_gradient_handler.py sha256=PeispW0UPYIRYlaTewdSfprSBhZRaYiRnYveAPXnqtg 1148
colossalai/legacy/engine/gradient_handler/_zero_gradient_handler.py sha256=EIfd3QXA9o5GPu1Z9SOAnZ55zLu0B2mzY6DSJBJkXFc 749
colossalai/legacy/engine/gradient_handler/utils.py sha256=vKmHFCyDLCnF1IksADtXtHz6gheKCoSqhrPwMtx9tL0 1020
colossalai/legacy/engine/schedule/__init__.py sha256=ti_-l2TTiGO-y4VzR4tmI5yrGxiAtJgycBsvE4Dnmk8 315
colossalai/legacy/engine/schedule/_base_schedule.py sha256=q9VDceCQ1vhcYO-82LRK9pWPFP47tn3kfQ7eiY6n50I 5816
colossalai/legacy/engine/schedule/_non_pipeline_schedule.py sha256=HjJG-SnEHIB24njETVZUgp6Ytfwzo29fcErEwA3qeB4 3808
colossalai/legacy/engine/schedule/_pipeline_schedule.py sha256=3h-tpk_3zlgjgkNWIUIoviOBycoGModtqHccLM9w4cY 40085
colossalai/legacy/engine/schedule/_pipeline_schedule_v2.py sha256=P6BpEljZrv1QmUFdTjFZB1JU8wc1vUtX43NzahNca8c 7133
colossalai/legacy/inference/__init__.py sha256=qqyNrz5zfQ-DKqUsrlwJxiieY6yObQfQe6tdxoWuLAc 152
colossalai/legacy/inference/async_engine.py sha256=iZTfUgSUnJVsGrmRZFq5CIc1LD5Or45vApBoP9xE7jk 4606
colossalai/legacy/inference/async_manager.py sha256=gK3uJbpW9wCM4W5sXjo4v0ZemBWa38-6BNUMP9LdEr0 5935
colossalai/legacy/inference/manager.py sha256=InsGwj5OEfap8ziVC9iU42H01xUfpSs1TKvlBVM8Eik 11605
colossalai/legacy/inference/dynamic_batching/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/legacy/inference/dynamic_batching/get_tokenizer.py sha256=6YC0jLCrt_48UE92zTKX18w1l6IxC5q9hlYMfAALqmg 1345
colossalai/legacy/inference/dynamic_batching/infer_batch.py sha256=1xERXCrH0Kg5gukRhZbEhR2T8hEyJGSPOSbCI2huiPo 13595
colossalai/legacy/inference/dynamic_batching/io_struct.py sha256=_5EjPcpkr0UnJFfJfXK_IKO-bTaLSy_irFo11C82tdw 5332
colossalai/legacy/inference/dynamic_batching/ray_dist_init.py sha256=rBbdYGNAivCDwnhwJEf8kO_7_Rn9gdnIFfGekPW8UQQ 6315
colossalai/legacy/inference/dynamic_batching/ray_init_config.py sha256=JqiqarBjhnC95TjiNG0ZMcZqxgHiXakJvccHJDnlNyg 1612
colossalai/legacy/inference/dynamic_batching/req_queue.py sha256=d_0UyFb90tdpHsmZzPL_Lm3hwU9HzHukpfz15UY937w 2810
colossalai/legacy/inference/dynamic_batching/sampling_params.py sha256=ErquV5yQZTBqRTwlXZ9HcySNXMtVZ8h6FBrF40dfp0Q 3282
colossalai/legacy/inference/dynamic_batching/stats.py sha256=D7MUQx5G6RW31ejdFxbIQ4QnXOGtoI0r7LJk27Ge6EA 1514
colossalai/legacy/inference/hybridengine/__init__.py sha256=bcNdbZDOEVdxqSfeLkiw8UfFzRkDvVK7q9la8QH0_PU 65
colossalai/legacy/inference/hybridengine/engine.py sha256=ojguulN5Tx2_omp9Z0tOsFA13m5PUIHbhrSjeiVFmB4 6981
colossalai/legacy/inference/hybridengine/modeling/__init__.py sha256=b8voUaBeDZ-Yu0lsVLahfMGP68JN8IQUnXp2z2nCbBk 80
colossalai/legacy/inference/hybridengine/modeling/_utils.py sha256=2MsD0iAGx8lFbdsbgvxBvMu4b8AANgdqCf2X1DJ_wvM 2706
colossalai/legacy/inference/hybridengine/modeling/llama.py sha256=38y_J-POrbEBvb7dBdi90WVENCSbbmjHvv-SiRSt9e8 21591
colossalai/legacy/inference/hybridengine/polices/__init__.py sha256=OZFyrDvEnBuL0ibAZB_CF33Z6HUVeP8DbOIr1PFmBS4 78
colossalai/legacy/inference/hybridengine/polices/llama.py sha256=CixiXFIOyl3bce5bKkmbT7rOQ_miv8NwDj22Ifve_QQ 5477
colossalai/legacy/inference/pipeline/__init__.py sha256=sRzrzgkCVqVjeIJon3M-u0MnTmsmXcJDMwG3YeHKobw 83
colossalai/legacy/inference/pipeline/microbatch_manager.py sha256=SakKXXcVyBF_VNIa8Si3rXbKIJ_5qA6jtcOJIP8wtmc 8993
colossalai/legacy/inference/tensor_parallel/__init__.py sha256=KinQ_W-jraR_Gp0WRCLw_IWrmY9c4iwpRZtx9hrCRFM 123
colossalai/legacy/inference/tensor_parallel/batch_infer_state.py sha256=Brchl_focDqhgUEfc2Ny7wBuQIbUQxNI3DDObnqETG0 4483
colossalai/legacy/inference/tensor_parallel/engine.py sha256=kmyVnFqifirzuv-_vGC8ub4LOePnTVmRWzbMmG5w1a0 21286
colossalai/legacy/inference/tensor_parallel/kvcache_manager.py sha256=Sd0wxT0-aEJbSqxmb5eYQBDaoHUYRDLcdCXZmYNvtUg 4580
colossalai/legacy/inference/tensor_parallel/modeling/__init__.py sha256=-pVey6YpPBfVGU-BCZHP87vGy6lmDpTz-cSI8GsaZPE 225
colossalai/legacy/inference/tensor_parallel/modeling/_utils.py sha256=2MsD0iAGx8lFbdsbgvxBvMu4b8AANgdqCf2X1DJ_wvM 2706
colossalai/legacy/inference/tensor_parallel/modeling/bloom.py sha256=bQECKM5boqqDkTCRSTv4TqgEYKqBpDNBen68G9JwKvI 23642
colossalai/legacy/inference/tensor_parallel/modeling/chatglm2.py sha256=ybUFac_iRjqxC1mf5EnGqYccfChumLLq7IxeYyBCuFw 22757
colossalai/legacy/inference/tensor_parallel/modeling/llama.py sha256=PGw0Yl-XEA4f9BtStv_z4rTzmMsV8IuDH_sQGB88Ldo 17832
colossalai/legacy/inference/tensor_parallel/policies/__init__.py sha256=wPeFKfqUzknFxAWiaH8ZUlj53bx2ZlQa_qG8PyOnfE0 209
colossalai/legacy/inference/tensor_parallel/policies/bloom.py sha256=8l3wo2jvgrsF7iQ69c86d1NzXDhp-KdH22E0xY73eyo 4440
colossalai/legacy/inference/tensor_parallel/policies/chatglm2.py sha256=KavGHP6_EKQNHeRwo1GGaRmNo_OJY4Yh8uiSrntlUdQ 2987
colossalai/legacy/inference/tensor_parallel/policies/llama.py sha256=CFSs-iwKZ7uiQX93QmMPBb0PIjyiWL0oJLDn3FTU7aA 4898
colossalai/legacy/nn/__init__.py sha256=NkQCspwuYd1hyf6pCCOvqAhLGtstXVEYHpKBrWv_kq0 63
colossalai/legacy/nn/_ops/__init__.py sha256=NoO8olEFA4k0OQUJMlJAwChSlAGDCiPrQocr1GdFsNE 22
colossalai/legacy/nn/_ops/_utils.py sha256=EFA9Wu4LCduk1584WZkFYf1KciQvrkV6z8BDriPnMI4 8669
colossalai/legacy/nn/layer/__init__.py sha256=-WIXwpqGPGYy_5ht5-egAdRbbylCiHAuLpBaKp96aI0 242
colossalai/legacy/nn/layer/base_layer.py sha256=x_z7AjRC0pZ7E9AZOLxzwuVWH_mFb9Vbnxy-So5CFRA 2817
colossalai/legacy/nn/layer/colossalai_layer/__init__.py sha256=rb7LqFyoZ8nlmSZDdNVrCNYGW_r-qyJAyVX3eix7_ng 300
colossalai/legacy/nn/layer/colossalai_layer/_utils.py sha256=6-fMCufX40GGIE-ixi3X9xbIPEUGk3en1VJSDyUflqA 1404
colossalai/legacy/nn/layer/colossalai_layer/dropout.py sha256=CjGpxk6ubo43C9YFEm7VsRHAwg9qcIOZvMp-Y_KoQp4 999
colossalai/legacy/nn/layer/colossalai_layer/embedding.py sha256=sHEcgNA_eEmS3AAiJKSJ4T-4l0_eymalRXAu_01vuBc 6156
colossalai/legacy/nn/layer/colossalai_layer/linear.py sha256=vyxZoHq--kIOC7_PqupGZXo1YmBwuydOI97DXH8afbg 5224
colossalai/legacy/nn/layer/colossalai_layer/normalization.py sha256=zI38NlarX-DMgLvs3uajjkWmluNd41K6CTcsjcD0mXs 1718
colossalai/legacy/nn/layer/parallel_1d/__init__.py sha256=BtrwfKlqtDqxRvpae3eWBfVy2MqowJSqj2x3-gHgqdE 459
colossalai/legacy/nn/layer/parallel_1d/_operation.py sha256=IqhWfweHXRpcHUbnB_84PUX30u6_1h5sgnOWjjOmgb0 3724
colossalai/legacy/nn/layer/parallel_1d/_utils.py sha256=v_MMK0mraNvaODu_48XOW5hQkQiN3xxXvHGXgRb8hn0 5063
colossalai/legacy/nn/layer/parallel_1d/layers.py sha256=PspGP8EHMQR-9hfv_yaDPzPyqYJXnQMCRo78AvvHMjs 44898
colossalai/legacy/nn/layer/parallel_2d/__init__.py sha256=Bm5m5UEJwks7oVTcsBUrH7ynniN3oFCbix7gGYHG04Y 458
colossalai/legacy/nn/layer/parallel_2d/_operation.py sha256=C2c_4iTUvMCP0BRbEc6B-mJA5eDtyJLuebEvKOLjGNU 35857
colossalai/legacy/nn/layer/parallel_2d/_utils.py sha256=vgPIxG3NDYnv6B8KdoCQ_wOsJOpSyEDUiMecZDHdHeI 853
colossalai/legacy/nn/layer/parallel_2d/layers.py sha256=hxSJ75p6lStBpMIJ6olRKXsURTW0h7STFGsQXjbQ0Vg 49324
colossalai/legacy/nn/layer/parallel_2p5d/__init__.py sha256=PdKktO8WaTGZaj3PZlCnTC7FTajsU3aM-LI_qSrGWDg 494
colossalai/legacy/nn/layer/parallel_2p5d/_operation.py sha256=LzARy0ZdiqQCtbp0h4faSS5JllvgishMUUhIluTIWoo 39868
colossalai/legacy/nn/layer/parallel_2p5d/_utils.py sha256=lQec_nQ2zbNRr490qo910tqcs9v9F5NLkjmCUwBxXik 1234
colossalai/legacy/nn/layer/parallel_2p5d/layers.py sha256=1g4pKvohEWbYQNrKNDZZvkAhp_5YrbQi_m5k8O5Cl1Y 49527
colossalai/legacy/nn/layer/parallel_3d/__init__.py sha256=-_b2gu09T-W_o_MH5w-e1qp-CDBOfnC0E292VdaCudA 498
colossalai/legacy/nn/layer/parallel_3d/_operation.py sha256=valcUonnlW1h8jiO499cAFpI6PK7MWhz-A5cQJzytRw 22832
colossalai/legacy/nn/layer/parallel_3d/_utils.py sha256=akvMT1cebw9iAlyUwpEXvDeb5KZzQhCd0nrEeMQnxdM 3037
colossalai/legacy/nn/layer/parallel_3d/layers.py sha256=cAiU16PvxeQDp25JnsSALyRZgJxaeW1Z7v5GA8x0dyE 49624
colossalai/legacy/nn/layer/parallel_sequence/__init__.py sha256=4QMjg7qadePEczMrHNgbQoVTNG551kLBJMHKQo2sI-E 152
colossalai/legacy/nn/layer/parallel_sequence/_operation.py sha256=tOo720oCIuaSOmQjoX9ug0ZT-HidQWdLtkxlnB5PVPE 6392
colossalai/legacy/nn/layer/parallel_sequence/_utils.py sha256=6gmVsEdnQbWDf6L5NLY4qxEc2Wr3qNe1YUkl7G7mnCo 483
colossalai/legacy/nn/layer/parallel_sequence/layers.py sha256=UVjP3tR_KwvXj2g_RsmBgoqDD4MEYzC6KmrOKCfIXZk 10693
colossalai/legacy/nn/layer/utils/__init__.py sha256=yC6atwbvBa1ZJUKfwYmoHTy1DU01ZrN0GyluVPqFxys 445
colossalai/legacy/nn/layer/utils/common.py sha256=QIcqoKqLz0Rie0tgsQjPrqWEcusiLj2MXhgNYft7yj4 2411
colossalai/legacy/nn/layer/vanilla/__init__.py sha256=_BRahV788Z1rAUJdr0oAGpEkTgzpj_ZaHRG3aeVVXOc 345
colossalai/legacy/nn/layer/vanilla/layers.py sha256=X2SOOBkZibqGPLoEObgNiazIXQa2oFVBUhmCxRtpYaA 14693
colossalai/legacy/nn/layer/wrapper/__init__.py sha256=oJqEjekdDkVG9asUSwOuG0WuRV-IKJNEcd1jXygTMTw 101
colossalai/legacy/nn/layer/wrapper/pipeline_wrapper.py sha256=fRZEXERgf2Iev0QQSxXkmMvI4g0G5yC4iFt3_4DkAIY 2170
colossalai/legacy/nn/loss/__init__.py sha256=0pSPNuktoW_qNENOgIRYjrEDxwo-Fxw_gUitNuouets 1592
colossalai/legacy/nn/loss/loss_1d.py sha256=5ViVjf-OU_WIPTKI1jgxCOnELslJLVNCchpYqMvRkco 4606
colossalai/legacy/nn/loss/loss_2d.py sha256=kNb7hy8FOQqtusGtBb_sEzYfMjxHcB4VTPJ-3LNGxk8 5732
colossalai/legacy/nn/loss/loss_2p5d.py sha256=wkn6NXkaM_5hAMMsDUPp0zLlw57lslXZlJkKiY9UvPw 5540
colossalai/legacy/nn/loss/loss_3d.py sha256=wrxLbg4xzCHcJ93rzRo7pmNOd_q6n1ZwUYq176SZg9E 6436
colossalai/legacy/nn/metric/__init__.py sha256=3oGN4TBgJI5YpNMLfdtu7e_4PGWdFD0RHKuXaWyq2eg 686
colossalai/legacy/nn/metric/_utils.py sha256=P4ukd0RQKAybH2MrgqvM7GMk3l7pv1ms4pftOP09SlM 148
colossalai/legacy/nn/metric/accuracy_2d.py sha256=TPkYWVnjFxE5N4XEIkGQTI8PkM6FsKN9XCr0wZo8KcU 787
colossalai/legacy/nn/metric/accuracy_2p5d.py sha256=ansDQm5qaCGbrs3yP3JbiqHtr5kjig2QHPZZV-P-Fm0 801
colossalai/legacy/nn/metric/accuracy_3d.py sha256=3-vXk0vGux2YXLFqIuaeJuZvRfOXDNlN1b6KcCPbvWM 1263
colossalai/legacy/nn/parallel/__init__.py sha256=k8UQbfaWyV2mWrb4coPvXSILGeqDjvsmFIvWjoolT1o 65
colossalai/legacy/nn/parallel/data_parallel.py sha256=FFeIK1ou6mVLZ3krpqLYClNsrGffDFqnJ3aFdawhHg0 6469
colossalai/legacy/nn/parallel/reducer.py sha256=m7_86gR43hsmVdNf2L8Q4Wg665oR-WIzWGrkLuuOvSA 3874
colossalai/legacy/nn/parallel/layers/__init__.py sha256=jfVv0cY-uSkV1u7jaIXjLl9FhE8xnF11-y82oTZcgic 962
colossalai/legacy/nn/parallel/layers/colo_module.py sha256=fWjTFqbe4U_jiS5wphEzpvAvpkDdo0EgIJX398N3U7M 1954
colossalai/legacy/nn/parallel/layers/embedding.py sha256=jmUB47ashQ34YQRbimo4V7hN215jg_H0u2XNbX5vA8A 1150
colossalai/legacy/nn/parallel/layers/linear.py sha256=8uc37X4G4m5sQ7mEzAHvYsmuLTNsROjKcK9S3RJFfo4 1136
colossalai/legacy/nn/parallel/layers/module_utils.py sha256=kQlCVLeuYViuJQojvO0xtQ3idYrC6F7a4te92YpFf_M 4780
colossalai/legacy/nn/parallel/layers/cache_embedding/__init__.py sha256=k9q9fYZxP3XPJaVyq_xVHDekvZAZPbVRBTf6yZToRzM 742
colossalai/legacy/nn/parallel/layers/cache_embedding/base_embedding.py sha256=Kwmxpcb1d9-TyoKcyvpVlL1wqUEXO2RDXcATCf3mwO4 1165
colossalai/legacy/nn/parallel/layers/cache_embedding/cache_mgr.py sha256=EwTxiKZZ5gptLFi0EcvTqwizqZ8Yn4DEc6R60Uhtxu8 27958
colossalai/legacy/nn/parallel/layers/cache_embedding/cached_embedding.py sha256=5XwbhlcBQL17iLy9AS23ovWcvYb5Qvhbtol-d3uZanU 8710
colossalai/legacy/nn/parallel/layers/cache_embedding/copyer.py sha256=7-2Uf4CowLoBFWMmDoZxKzEumILkFIsmHxwg-al1grk 2041
colossalai/legacy/nn/parallel/layers/cache_embedding/embedding_config.py sha256=jAjKit8JtXksHEsrL5_WdE6vnqsmhDnIKnPjkw43cVg 807
colossalai/legacy/nn/parallel/layers/cache_embedding/parallel_cached_embedding.py sha256=fo-2cpMHJWPnZk862ZtW61Sq7m_-i2f-WwQ07Ryl5-U 5690
colossalai/legacy/nn/parallel/layers/cache_embedding/parallel_cached_embedding_tablewise.py sha256=7ulba8xVYFhq1DNvLm5Cg4Bb-O3NjEdqZOkUdSBRxfs 9962
colossalai/legacy/nn/parallel/layers/cache_embedding/parallel_cached_embedding_tablewise_split_cache.py sha256=MP1U8Mua6FDUUIOOwqxlsohnjCUPuawmK4L-Hassi7o 7138
colossalai/legacy/pipeline/__init__.py sha256=5N3Wz4GrZ-jjnHhnlgyTYkQnoVhtIB6VR4n8PHktyqE 163
colossalai/legacy/pipeline/layer_spec.py sha256=HHqDBG4Zzp0b72uhRXj4NI5MZGJfDM3o9RfcclWAzCI 1558
colossalai/legacy/pipeline/pipelinable.py sha256=uRflXU3HZK6cddq8u2oN1f9thbF4WNVlaesG7qgpjmU 11447
colossalai/legacy/pipeline/pipeline_process_group.py sha256=PdKQPBi4Z_bu4bKYTjXe15qQ_fpHhyCW0p_KEUjAoFw 5759
colossalai/legacy/pipeline/utils.py sha256=tT3Gw62C62F_XrrO5vMbkgXfNbuT3iKDSipXQoFkXgk 9017
colossalai/legacy/pipeline/middleware/__init__.py sha256=wGYPmpyDpCewW_VKSoPjJ5UFrnaLxCCWWNrbr-O-CBI 149
colossalai/legacy/pipeline/middleware/topo.py sha256=kJ5YGPvU__6E-TYJ_WGGv67fi71POIxpN3Ms08Es4yg 6818
colossalai/legacy/pipeline/middleware/adaptor/__init__.py sha256=xlxuQ49wlJK5M5ZmtpLsDIHgaGNvLqsCBl6lh86XP30 79
colossalai/legacy/pipeline/middleware/adaptor/fx.py sha256=efIJsBFroUQ0tlxjHwOZH1f6CKBKd3mIHjsJ67mY3jM 6151
colossalai/legacy/pipeline/rpc/__init__.py sha256=nqbgY3cOxebdLeB2PgZATEaEud16iQ9Fgu4COwf0og8 237
colossalai/legacy/pipeline/rpc/_pipeline_base.py sha256=8-tfwMyWzXkOE5N8f1OvNUhHjniLiwWNVqfKI5tVY-g 59091
colossalai/legacy/pipeline/rpc/_pipeline_schedule.py sha256=R-X_uNoGzgBtv_CxYFNukj6pCz06WR1y24h8wglkc9w 14979
colossalai/legacy/pipeline/rpc/utils.py sha256=LEU2AUOygj2QAftI5wTCSk1T9kz6K4T-tkEVbYm-Et4 5390
colossalai/legacy/registry/__init__.py sha256=1PFbdbULxj2vCNSxgK4fckqgcs4x15kBt1K6EA3FFMM 690
colossalai/legacy/registry/registry.py sha256=l735HLTSd4GnEefwtz7bhP1wAsDFZK0QPIS_HN2E488 3052
colossalai/legacy/tensor/__init__.py sha256=H6Q8WKsUbNIkmUkG7JL6AsGz-lJRPxJLOho2NgcbLXg 418
colossalai/legacy/tensor/compute_spec.py sha256=dAzvuXCsqLa62m0QKw8RS_gK1tm3hNSic_BKnFPQA_E 779
colossalai/legacy/tensor/const.py sha256=oOMEijGOKFKBvsvZQb6ARAffcOh9Lue3MkeSiarJbv0 101
colossalai/legacy/tensor/dist_spec_mgr.py sha256=H8asUDYmnd9q7tF1KLLfrJ_vFeQlBlmRlxmT4zcPYt4 8692
colossalai/legacy/tensor/distspec.py sha256=3CV9jKlLsuxTIr7rohAArfOGXnyxazjopr6JDSHCH1g 2714
colossalai/legacy/tensor/op_wrapper.py sha256=ThqGGmeXRoAm-03z38dV7d1JYH3sI1BXPzC16W_Khjs 1678
colossalai/legacy/tensor/process_group.py sha256=K6lxUoJ0OC3by58SEmAwwhuQTHeJ6ow1VI-9eFen4zI 10505
colossalai/legacy/tensor/tensor_spec.py sha256=mSuaT8slmCQcpI7k2petVZTWg4R-GiLmXwYf_y-m_QQ 697
colossalai/legacy/trainer/__init__.py sha256=eo0OVyVd3Pf0lXTVJ03apw3ydaE2fqIs9Ku-CT7x3Y4 53
colossalai/legacy/trainer/_trainer.py sha256=cNfE-VQWzYhg0r-xtrbBN9R33zv6FEW5B78xMiOdU6Y 14773
colossalai/legacy/trainer/hooks/__init__.py sha256=-YekSfVvoza9LASLNWGBR1JHvKWaw0KgA7-87MDxyU4 648
colossalai/legacy/trainer/hooks/_base_hook.py sha256=ccDdPEADGT5JfLfxSnnlf4hNH-dTWGFjOIPGJM2jlKA 2712
colossalai/legacy/trainer/hooks/_checkpoint_hook.py sha256=LLg-UK1gwBD4LoFHOMM7hJANX1MHZxqP8N3pOpq4HpU 3226
colossalai/legacy/trainer/hooks/_commons_.py sha256=_LhHaOFcCF6k3zP0gMESXDCc59FIJNHMdsMk6a_w00I 231
colossalai/legacy/trainer/hooks/_log_hook.py sha256=QIGrbvkWc8Ax9twOiZ4FRWK1jQbfYveZ-MCMXiLt0rk 13085
colossalai/legacy/trainer/hooks/_lr_scheduler_hook.py sha256=HwBfsj1NkDyldfd0JtxJGS49CuakJzzNA0A_gPe-OHI 2091
colossalai/legacy/trainer/hooks/_metric_hook.py sha256=Mubq-6MxN6qZ1JVCD-oIJICoTdGQ58VXdokZbYpYST0 16242
colossalai/legacy/utils/__init__.py sha256=5xlQn1OJ-ppGAl8RqWM95RzibbfTNQXespEIafdZl2g 1438
colossalai/legacy/utils/activation_checkpoint.py sha256=8Ynl7KFtZOFraz7aaT6KnNTg7D7Ngv_Gw2cExot8YN8 9872
colossalai/legacy/utils/checkpointing.py sha256=wavlAWxSDGLjRNmzDtpHmB5eHlcPyY41EUEF__pkszs 11338
colossalai/legacy/utils/common.py sha256=MBFLE_T_Yon5_n-lS1c5SwHWGnFTAA8pII1vwfs0w_Q 16595
colossalai/legacy/utils/memory.py sha256=FtnEhO2s-k-XSBuqt1E4x1FmzywG-e-I29b3lwypU2k 6416
colossalai/legacy/utils/checkpoint/__init__.py sha256=OjNGxT8hscVNG2TIrlg9tWDttrBzZEyrl2eh0yxDzsI 114
colossalai/legacy/utils/checkpoint/module_checkpoint.py sha256=UmmM0y0i2JU4glQUJA88Kmc2DgXta0qsd11m-rTkD1I 5297
colossalai/legacy/utils/checkpoint/utils.py sha256=Eudvu-OjnWbkvMgL8T-6irxU051C7YY7d0M4DbAtelg 2149
colossalai/legacy/utils/data_sampler/__init__.py sha256=vou3KBdRRbUZfzm0rQBRc9EWS8OhcrH2BxJnM9V_Zes 177
colossalai/legacy/utils/data_sampler/base_sampler.py sha256=W8TVq3JctZ0V41Y-JT_IgzrAZhm70yPAoaZR1obhTs8 339
colossalai/legacy/utils/data_sampler/data_parallel_sampler.py sha256=XmL8Rq_btP6tl-ZsX4xkjln5xwPuE36llpszbhUr1jc 6704
colossalai/legacy/utils/profiler/__init__.py sha256=sgSx5c6_NRjv_3pmz22fREeoD4o0OeR54z5z_-9wmbY 52
colossalai/legacy/utils/profiler/extention.py sha256=ifa4JIXwOaRXKrSqir_kxEP91pMGP0BmqqH5USN4kPw 341
colossalai/legacy/utils/profiler/profiler.py sha256=NDXYgUEuSsEM3YbeOBrzCEqtfrHXcsrMPdprAnzT0Ac 8459
colossalai/legacy/utils/profiler/stateful_tensor_mem_extention.py sha256=cgKqbYoU5KwowVWHysyF3A64zflMVKm63qf89JjJI-M 4036
colossalai/legacy/utils/profiler/legacy/__init__.py sha256=iOv87OAMtTEDTDinSf7iwo9oifR3cqwn0njFpWKdwyM 266
colossalai/legacy/utils/profiler/legacy/comm_profiler.py sha256=ZKpn3w3x2ytJIPvw3tu013-LmtdpjMow72f1Kzq7VJo 10461
colossalai/legacy/utils/profiler/legacy/pcie_profiler.py sha256=4D88jkrES1D08asL-k8Z4rDl7PtBILF3bDKtvpIZ-2E 4861
colossalai/legacy/utils/profiler/legacy/prof_utils.py sha256=_gjhTPOXBNMRmEk9XNgRX5TKdzpKqPcDs4WhcPMYvJM 3776
colossalai/legacy/zero/__init__.py sha256=yiYmFmOQz1joeiLSwV3M2atL49wvkBTot8vRFKrvp0E 1570
colossalai/legacy/zero/gemini/__init__.py sha256=AH4BMpiP8Z6IC4IMRqLARX6thEyo_65Mt0zJqs1WjE4 619
colossalai/legacy/zero/gemini/colo_init_context.py sha256=Dz_koDH6qIC-kEeUFSU7WQv9ucj7sgFRBw3W8-bmtwg 7315
colossalai/legacy/zero/gemini/gemini_context.py sha256=cj10R0orKcmVC5wl0F-nMgthPNluFn9ZlzL5BEwyTu4 1518
colossalai/legacy/zero/gemini/stateful_tensor.py sha256=V5jBZtmm8-R_IuGO60T50JYACZLd1_gcw8Ys3VS_05M 6905
colossalai/legacy/zero/gemini/stateful_tensor_mgr.py sha256=epR21EXA5oUvLlyTOI8Pu-D4s3Vyd01kgGqEQHRokBY 3965
colossalai/legacy/zero/gemini/tensor_placement_policy.py sha256=71VQbwV1bQiA-GkNSXngNuO6sFS6rGlESZan61SrnhY 6369
colossalai/legacy/zero/gemini/tensor_utils.py sha256=cd0Q69cYZuR-lUX_jDZugvqFno-9RPN0GRRlZO9i0us 3807
colossalai/legacy/zero/gemini/ophooks/__init__.py sha256=3TV188qunrOl6RCQCKsO55iCHX4VMkEvg1NCBhC4wAA 118
colossalai/legacy/zero/gemini/ophooks/_shard_grad_ophook.py sha256=DFwigIh4pf2k-IpDmcpU6pMXJF8Ub7DDlVTzP4l_huQ 774
colossalai/legacy/zero/gemini/ophooks/_shard_param_ophook.py sha256=uh3tTEjJYKeVjGCkct_VgYwP3WBFCnkgLx6X0ZDNIro 1363
colossalai/legacy/zero/gemini/ophooks/runtime_mem_tracer_hook.py sha256=vC7HmPMiqF4sBpeRYqh_--wrdc_ajgHGXtc4DUsR3A4 5081
colossalai/legacy/zero/gemini/ophooks/utils.py sha256=uDhQTHSnAov-7tlHN4dxngEapDnPMPrGRnNYBDXBJj4 4715
colossalai/legacy/zero/gemini/paramhooks/__init__.py sha256=ptYuAXGMfcIMkQbc5r0H23ttSn7bBck8jsGucUFcoWo 77
colossalai/legacy/zero/gemini/paramhooks/_param_hookmgr.py sha256=KisD5Lh8c41WPw5CnpnfRZunI-yYO4PWqZFhLUAUuV8 1251
colossalai/legacy/zero/init_ctx/__init__.py sha256=rJStHNDu37hYRZmd4HubGL634sVlyV_R3qJUk4Byc7o 171
colossalai/legacy/zero/init_ctx/init_context.py sha256=LVL365EPe6-c_dtr_6GeDTxkdc-0hTBp8IfCzX6vmGM 11099
colossalai/legacy/zero/shard_utils/__init__.py sha256=rG1qMThyzoEwGyThAarzxF5v7mu5iW3FbC9eTSHSuh4 259
colossalai/legacy/zero/shard_utils/base_shard_strategy.py sha256=0a0jRmfpWqozhD17w5pKfRSQ7Ot4wrWHT2cDsPKSTL0 635
colossalai/legacy/zero/shard_utils/bucket_tensor_shard_strategy.py sha256=4dhiey0XaJwKx7--e95JkkXEPfZvPMulPsXKB55cF7s 2296
colossalai/legacy/zero/shard_utils/commons.py sha256=7P7dR1Vb4ho0qAw8seNzEdmrF3O_cdDYM-_ixwVs3EE 706
colossalai/legacy/zero/shard_utils/tensor_shard_strategy.py sha256=tzOBLk35GUAnXjCNNqhXaoX4hmA_3dI5OhHLxdvlRtY 2751
colossalai/legacy/zero/sharded_model/__init__.py sha256=ItcIYuI3nhz_ubP9reur9ghMg6FaUIrMFa1Y3ryD8lk 75
colossalai/legacy/zero/sharded_model/_utils.py sha256=DDggT9z2hTRqHzYUqVE_7uzME5X_2iIDcqSwqQGTheE 3003
colossalai/legacy/zero/sharded_model/reduce_scatter.py sha256=fo4p5pi0VPowc3j0sJqPvtTe5Hfr2dPh9ax2zo4QHuM 8289
colossalai/legacy/zero/sharded_model/sharded_model_v2.py sha256=hdv68PvzBBGVhJ8VOP_6ljtYBJkUkTl-swEVKalCF_o 28757
colossalai/legacy/zero/sharded_model/utils.py sha256=4ABWTVsdXo8fyAQjUTB_EjWjkKFiITrLCSPpY53CL3w 808
colossalai/legacy/zero/sharded_model/zero_hook.py sha256=JTyVm-vfS4A3J61TlYr6hDcez1zyZwozSaDjuowulzs 4865
colossalai/legacy/zero/sharded_optim/__init__.py sha256=8iV8fryIjMtsVBnmryr6bhpNTW8ebp1A7iAde2AP1QY 83
colossalai/legacy/zero/sharded_optim/sharded_optim_v2.py sha256=giZgst_LCvETTDoxkybQcPXh2lH2wSl6t_DPYb8UU-A 18982
colossalai/legacy/zero/sharded_param/__init__.py sha256=sPJkE2DlaMAaQosKRqOoocqTmHizYaoZks-LVd9cC_A 131
colossalai/legacy/zero/sharded_param/sharded_param.py sha256=fUTOpeAwg3Vcx4mh-XL01Mq0thHAJNKTp5py7SgxIjc 3859
colossalai/legacy/zero/sharded_param/sharded_tensor.py sha256=c2toPW1Uhp1dGYdnoLUWKzvGp7kRPG2vZD5w1MZMPI4 1144
colossalai/logging/__init__.py sha256=YVE6yxCOoPOmT627k_ePT-OGAy6drRTOTfjeN-qaoEU 1593
colossalai/logging/logger.py sha256=RI5XWpA3WUZJRJFBkE7F9XycDmclFlVyyt7M3vfZHVw 6110
colossalai/moe/__init__.py sha256=cpDUOwFTAHvveN1lnX221ODwUMjO-qWBudw4mJZ_CNw 529
colossalai/moe/_operation.py sha256=jYRv1ybna8BkyGJB4aof9buI6nnGV7RhLpsu5JsmJ0Q 12432
colossalai/moe/checkpoint.py sha256=c_Szfz6htTT-2iyyXKd9ZvDFxDBHIFDGRavXaoV6_y8 36074
colossalai/moe/experts.py sha256=yYkRRWG9HkeawHZfRZlwbr6hgD6Ogv7FgrKW9Q_cI7U 6249
colossalai/moe/layers.py sha256=dvYucECRhb1xon8nZmYqGBNugn80cuIWSw4CSIXkchw 16261
colossalai/moe/load_balance.py sha256=3ziEoCKbkCqcNFeNaxendDTA4jFf2wGV_DsmXwBKt8c 18267
colossalai/moe/loss.py sha256=-DGbXPFqoyK2K34ssLGIPqXl--scWGIB0JZozgjM3UI 3075
colossalai/moe/manager.py sha256=2P21tHuX8vK7pgc69MCNVl04D6Sp0FQvRolftk_1acI 6138
colossalai/moe/routers.py sha256=hdSIjgStvBZwiez-_zMyl-QuIjzx60fxPTPqgtYLrtg 20401
colossalai/moe/utils.py sha256=QuT6Za1_URfbJ1ejEhgxlb_7kSfTukI0mrXKZu4iEB8 7403
colossalai/nn/__init__.py sha256=YAutugOnzI_Jqb5yDutTyChbtA75RvZEHg4CUErPVUE 114
colossalai/nn/init.py sha256=SMAcyx7un75s8yxttGz1StRthLWu2aXDWvDljJx2tK0 9563
colossalai/nn/layer/__init__.py sha256=alIDGBnxWH4JvP-UW-7N99seBBi0r1GV1h8f1ERFBec 21
colossalai/nn/layer/colo_attention.py sha256=szZ_7GiLk2aC-y_u4t7aIfN6CBlLTe6TBa4n0Ifdfd4 7635
colossalai/nn/layer/layernorm.py sha256=4ETxuxbkUHCWBRH7y5qRE22z6aqffD1npRLliRW2Qxk 2497
colossalai/nn/layer/scaled_softmax.py sha256=3jI7CNEFIr9YX6MeXcZmMRVdnN30ZP5-RGvFIqB6VPE 6394
colossalai/nn/layer/utils.py sha256=uHBvY3595oRr8ajLNuLfVDA3Ikv7vAlwNM4eeua5CO4 449
colossalai/nn/loss/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/nn/lr_scheduler/__init__.py sha256=Ab6Rdz4jHQbK2AUv2KBAKf8MptI8O9zo74HpXwJ6udc 673
colossalai/nn/lr_scheduler/cosine.py sha256=aiW3r9YKLPSujbvPW7asShmZCeTnE_mrG1ZXemHkjic 5867
colossalai/nn/lr_scheduler/delayed.py sha256=pjpNvoK738s0co__-zLE_UpijylYm2iiRxJ0_RKlptM 7717
colossalai/nn/lr_scheduler/linear.py sha256=C1ZCxKZJb7TSf6pJwkN9qEZjliClVIe-KgiFwi-TfJ4 1178
colossalai/nn/lr_scheduler/multistep.py sha256=0B-It5kQ_4WIoCF2aNZNdCCp9VKfM93waobeQiiNL8U 2672
colossalai/nn/lr_scheduler/onecycle.py sha256=zNw0gvTc41YLxhLHBZszc8hqi4-TIxFsuX5Od_82BfY 5050
colossalai/nn/lr_scheduler/poly.py sha256=QVYWUSWMUCPi4CRdWVAkf5nZuQgnyoA5n9MxVmmrbTg 2458
colossalai/nn/lr_scheduler/torch.py sha256=Ed1hY5Oxx-iQU5kd97n_4CyiMJkDYmcRTrVc0OS4ZUk 3516
colossalai/nn/optimizer/__init__.py sha256=gY9b_rcaWVCrhi7ABjTYuA6t_Ny_ICgY_OzXIAO4Okc 303
colossalai/nn/optimizer/cpu_adam.py sha256=k83FH6StPbQPPmD2hJq1W7lHEGrYqe96DymTiuSQ6mM 8610
colossalai/nn/optimizer/fused_adam.py sha256=QmML4j0ZQW-Ncq_h8jbsyMB4JSEYM3NexbEiGqxbJzk 6420
colossalai/nn/optimizer/fused_lamb.py sha256=I-7RrgD9UykD_uqGx034MgIAZ5tCZDzvv-VPG_Se8JI 9056
colossalai/nn/optimizer/fused_sgd.py sha256=lMzb8Lm-AfVqhMOpjqUTVDfGtQbu1TnOMzMCTiU4q5g 6012
colossalai/nn/optimizer/hybrid_adam.py sha256=_M3jauDrlAGynjWAl1p1n0XuKX8hbwTh_3nmMPQRA4s 7946
colossalai/nn/optimizer/lamb.py sha256=UP-I4UzOlu8-eFLZaH2KjkFsHLydv-o1KXkdAEDJr3I 4423
colossalai/nn/optimizer/lars.py sha256=FAI1igP7mB1dpNXNI05fGY2FpS0JvdkLOmkda8qRIbk 3567
colossalai/nn/optimizer/nvme_optimizer.py sha256=buMTIsbMiTQEnHnwDejq-zFcwi09UhEjTpxQZqxmZiU 6768
colossalai/pipeline/__init__.py sha256=o53u6F_X4W7gj00WZh4sbZRijKN12oIw5CmhGIF9RBk 344
colossalai/pipeline/p2p.py sha256=zJhNrCwz29IU9WNefQjEuTESYOHPbzCD-xcdSpLU0NA 26187
colossalai/pipeline/stage_manager.py sha256=PSnWjrZ5IUgjlCU-2_4qU9_Qhpd_hLcQ5VtfKYAUxqY 6790
colossalai/pipeline/schedule/__init__.py sha256=5b097OjmrewfMx6Kl40JWSTd66Uk14uOiymP7sRhF8s 241
colossalai/pipeline/schedule/_utils.py sha256=nbBt9E10u2GmByzcGA_310fuyXcSqWsn0Xd8c3VgdKs 5194
colossalai/pipeline/schedule/base.py sha256=V740bS4do6mWMjTfM8KxngX98GS_lUTd7fLT7rDIxd0 1478
colossalai/pipeline/schedule/generate.py sha256=OW9S0LrutDeTjVDsvpjqVMAPKd3JWJ-axi1eeSLVpzU 20001
colossalai/pipeline/schedule/interleaved_pp.py sha256=-6-1wgYpvvp0et5t8iUa1AySTDnfgD4hK5wEjihNgXY 26918
colossalai/pipeline/schedule/one_f_one_b.py sha256=cUzI9SI5fwOOc1w2VnmwbTF6x8BBLZOQNl-mFVLrQeg 19503
colossalai/shardformer/__init__.py sha256=BkAY64ciuCYyPwkzTfD9bve6Zpzd0RuFpJFQ-pLoRbM 44
colossalai/shardformer/_utils.py sha256=SURmpkl_-5ggliakR4N3_DObGqoH8LN08lx2FlkcI8c 3327
colossalai/shardformer/layer/__init__.py sha256=S0Ya_1nYzTOe23xwhxJSKgBebzWqfhfA-IZp2TGtHJw 839
colossalai/shardformer/layer/_operation.py sha256=iFJskZIPYIDVdHRsmBpQDbexYTayQHakl8fmybRHbSU 25183
colossalai/shardformer/layer/dropout.py sha256=ZHA5k-VZA4T0bBd4pBiR_iGZw1aC8YoQpTcTvukUWi8 3520
colossalai/shardformer/layer/embedding.py sha256=WR4hvp_4Vfe33apK-LR39WqwEqQDTd-EMSb8fgDrZFk 12912
colossalai/shardformer/layer/linear.py sha256=01guHv7cqYdjKPc3N5H5XGDwEku8zNFcy232LA5IjlI 17470
colossalai/shardformer/layer/loss.py sha256=t44dtJXNUj6Lhg5zt1pDvcp5dRG-wqeOvRj1uV6GNc0 4589
colossalai/shardformer/layer/normalization.py sha256=m38QBgQQn17BZtOxfjBmR6DBUWMowuRLNh2cFnZZyN0 11366
colossalai/shardformer/layer/parallel_module.py sha256=yDbnWcYJUYEj0Xv9pfIR98y4ZKrbG83GBisD_XXXrdQ 8403
colossalai/shardformer/layer/qkv_fused_linear.py sha256=zLOR0Z_So2riejzA2pJIq31lMA4v1_tBiz72oWNhVw0 31070
colossalai/shardformer/layer/utils.py sha256=jWusNrgFOFHpWsIYyzjcZXGU0PBSjm-YNnwjUL7oXXg 10310
colossalai/shardformer/modeling/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/shardformer/modeling/bert.py sha256=3rHVAgN9ZFf0xuCVzN8BA8w1WZhDPqLsbVPjhZ-_oOM 62017
colossalai/shardformer/modeling/blip2.py sha256=bRGgqCPkzSqeZWaXSyUDbmSzLimbnrbUSdYD9iRGgn0 4254
colossalai/shardformer/modeling/bloom.py sha256=WM8oovE4MYDcdqUlXJ5SChmMtRamyJHSRSYQUoUrohA 49686
colossalai/shardformer/modeling/chatglm2.py sha256=wAUpxtHeEwd_yWsRV9uQIaRatUUtJ6HPOGCd3sxOY_s 17492
colossalai/shardformer/modeling/falcon.py sha256=Il3jqrqXg-l7h34Wm_AzneJzsmIXy9S10wOFEm-wb4E 34661
colossalai/shardformer/modeling/gpt2.py sha256=asLfxs8IgoAF0k6k6LCVlJQDPvrrWSbaqVFWlDQF17o 47396
colossalai/shardformer/modeling/gptj.py sha256=JxDVHDKN8D03L_DWrrAmpFY1uJWx7o5MLfAte5LbjCs 37538
colossalai/shardformer/modeling/jit.py sha256=WBQtyV6YJy7VJoVUA7y6PI_9Ts8K-yI6C2QdadJpRus 1053
colossalai/shardformer/modeling/llama.py sha256=-lzIOyBMyQmtaPB_bRkQ6RkXxYO96M6WJizgFtKSJk8 27010
colossalai/shardformer/modeling/mistral.py sha256=KJaYv47jBaa3IboGe37EyMEI-Uc8a1brhvWznKeiIh4 3235
colossalai/shardformer/modeling/opt.py sha256=EYzjhO7BR6C4B8xzQkqRVOaNRZ0xNkWUVwijmh7Xviw 30438
colossalai/shardformer/modeling/sam.py sha256=93ev7VTSwTaIZrsxkaG7JYZ1gohAhz4d1DgbiQYa524 8328
colossalai/shardformer/modeling/t5.py sha256=wrmidckdcfc_cj0V6g-A5cajYzl8GG8yERkj8ymh3eU 37937
colossalai/shardformer/modeling/vit.py sha256=QMoaQikMxh2yAaZxtVV3zmRF37m1NkSUUJtTj0OXEEk 15888
colossalai/shardformer/modeling/whisper.py sha256=7UFvlo4gM47Z0_bRJIc_VHwtpJV6isUezG0Kx2SGQig 46618
colossalai/shardformer/modeling/chatglm2_6b/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/shardformer/modeling/chatglm2_6b/configuration_chatglm.py sha256=gRYsGi35Zoe54wl5Gnr4yaWYes23XQ3cw7NxRQNJXxY 2249
colossalai/shardformer/modeling/chatglm2_6b/modeling_chatglm.py sha256=PdnYwkbfU-jRqp27g-icN8b10QSbBOFlbOT6_X9-vi0 54609
colossalai/shardformer/policies/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/shardformer/policies/auto_policy.py sha256=482ERQ1Z8y3NHt5pRpGrGq3GRjtexuIh1xqMJZVMkcc 10382
colossalai/shardformer/policies/base_policy.py sha256=Ruhrn8y6dJa8Xfstiz64oTgjwYi3l4E_KbwZqPknyGo 10322
colossalai/shardformer/policies/bert.py sha256=2WZ_39WkL08vU9h9MkMMt2q6pBrwIdtg4DOO0EEAsyw 25384
colossalai/shardformer/policies/blip2.py sha256=olZuX4Hlcq_k-vby1PseaFV7QNWzyn8S02FjdddoU4s 14089
colossalai/shardformer/policies/bloom.py sha256=N1p_fQyGrfYGCv82ZRerp-X64GGtSPgJQNLeIfkdGTU 16402
colossalai/shardformer/policies/chatglm2.py sha256=0Sw79OyvoOdNr3Z4TbYDdSbUXH9AP9rcKBHOQfu8bRo 10609
colossalai/shardformer/policies/falcon.py sha256=SYuD7_RbRyUsNrYFGXI5bfIlSACQk7AntHgwKFCOS-k 15893
colossalai/shardformer/policies/gpt2.py sha256=4yFfYgUWgmKkWFQzC0N1YOBUlPxGNUofIUhxa9Pcr94 16168
colossalai/shardformer/policies/gptj.py sha256=aeItUiwtxHfQbRUKElrjnuMl0TFR0G-FnBLJDG9iFAM 12492
colossalai/shardformer/policies/llama.py sha256=1Sbg-HgTwV_3JoloMbeKciKRkV2tqyT81lmZ-gf_Xs4 13337
colossalai/shardformer/policies/mistral.py sha256=M9FiKYpDqyzf8njREl5xWpr1o1jaCWY3iGTIU22udsI 7142
colossalai/shardformer/policies/opt.py sha256=5P0k9m6wtf-7NzYL1-rxhaI8z0ob0ZN8-4oKNG46qXs 12115
colossalai/shardformer/policies/sam.py sha256=faVTa13SQ_95QHqWp3j3m5OtJMfiW_gbReoQkyrkLs4 9404
colossalai/shardformer/policies/t5.py sha256=52UESlMH7Efg-hw5rUqiMkHOO5fOOI4yzhAeQroeeDs 19859
colossalai/shardformer/policies/vit.py sha256=Q6NG_pfwj6qJczn5p3YmJjEx8ZngTY8oHKmAEJ_Axbw 10056
colossalai/shardformer/policies/whisper.py sha256=IRZaaLD_W3X8mSuRUdbcN4YMDIj8MXI7Gcpjm2oX0oo 21297
colossalai/shardformer/shard/__init__.py sha256=RwQeNkUuRrM080ndMdfKv_LxHoTk8WcECGmlilsyMj0 167
colossalai/shardformer/shard/shard_config.py sha256=gO_nZLVlazyzTAK3qNkeP1xz9gvNxxPov24aX6Id5jc 4007
colossalai/shardformer/shard/sharder.py sha256=DobC55PzEIJcaM1XWQrs_OSdt-WTKSkUHs6xJUvEXtU 9749
colossalai/shardformer/shard/shardformer.py sha256=vKJfe6pOPa3HTqtaK9Dd6kG2UVIK--vTT-Jpj5okSr0 1593
colossalai/shardformer/shard/utils.py sha256=xnEW7LBpoAcj5PqHKeaHaBeJTHp9q7vsLz7BqTcljh0 550
colossalai/tensor/__init__.py sha256=U0czdM80FII0j1_Q5x_B7jAoTyQ4IKhqnELwXkZZV-0 600
colossalai/tensor/colo_parameter.py sha256=pi9yjySdtuFhC0QjLRNCTDG53RyTlpu85382IEyWtzM 3447
colossalai/tensor/colo_tensor.py sha256=cCA3b_BhR3IOETA4RlzHGcVmKAHECHteD6eGt5voK2Q 3349
colossalai/tensor/comm_spec.py sha256=qMeflMBMhpox53xWmY2s-lxWN4IuZHmuBx7tRsl_59A 21633
colossalai/tensor/param_op_hook.py sha256=8MgYB_Fh0ubvYMXnxYFrC7ncxdyWg5wbi3KeRQod45E 5169
colossalai/tensor/shape_consistency.py sha256=YBmXDRwy-31kq1DpXZ9Rcr1eUy3nyqTq9bHWqZby65w 35882
colossalai/tensor/sharding_spec.py sha256=KvFo5mnf-pZkCgrZOgYJWPIVxbUiIWGnzPi66TKcSpo 11626
colossalai/tensor/utils.py sha256=fJgTRGqam48m-guP2TFGvyD5DQBngmfnY91Vp4r2FrE 8452
colossalai/tensor/d_tensor/__init__.py sha256=MXM9sis2ha7NKTMuomhCKE41Y9Si5NU6xeEOl3RgJcA 1203
colossalai/tensor/d_tensor/api.py sha256=EOcZkZt4Cw8tOH0-IieOJvkea25wKqKMNbgtf5NWIak 18027
colossalai/tensor/d_tensor/comm_spec.py sha256=SocjA8WXrwM8S0oiL52ONZwcNXfXNI3t2m0NhtyZ3e4 11144
colossalai/tensor/d_tensor/layout.py sha256=Vsiqt1KCCuBW9YcBuKiuzKBu47J1RQOrI3Vq99sIc1s 2776
colossalai/tensor/d_tensor/layout_converter.py sha256=DYzs9ttNd3HN6qnUHWUzMRN2O7I_3SpmIiUPCYw5joA 26974
colossalai/tensor/d_tensor/misc.py sha256=1hwxoI22qUWLZ4vEE4uXWNnUT8QGN7x-kkzkXf6hruI 231
colossalai/tensor/d_tensor/sharding_spec.py sha256=pDhgFyPqEpgIEYj2brKiq0U9iD-sG-L3erOCuMmKDDM 9293
colossalai/tensor/d_tensor/utils.py sha256=GWV3rgpsMMYx33TPy85usuThYgyT8o7GMKjtdoTIzB0 3263
colossalai/tensor/moe_tensor/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/tensor/moe_tensor/api.py sha256=7btejY4iphMUgd19WMNOWdN5tpiRHrFO92IIoa5lcf8 3778
colossalai/tensor/moe_tensor/moe_info.py sha256=2rKbZC40m5Z6vVPhl4il3Y8uDh63NYCh8OKkGJz175g 1413
colossalai/testing/__init__.py sha256=2TerCo_zrotqPt9LuQyFN-l2OHzgFNTUFpGVzRXxg6Y 865
colossalai/testing/comparison.py sha256=GYL-16EO4doEk554ZrTTDPRmWQ96J51HrlsC-V3-s10 5348
colossalai/testing/pytest_wrapper.py sha256=zaQ8yCcPTWItUUWioeUcfE5HuN_r9MRiHEj9FaJ6gbU 1406
colossalai/testing/random.py sha256=bXP6WA0ZdnMFOppRDUAHLUW4e4XyaMVIKUi1zEw5Mz4 542
colossalai/testing/utils.py sha256=kf-5ezLTfD-DdM-YMxL7K5kPJ1KQ37njmN_gWKXInTY 9201
colossalai/utils/__init__.py sha256=VJKq7KVR2dPC_M7aQpa9kNwm7E5_zh4eUuyoFh8tuSE 586
colossalai/utils/common.py sha256=beCK4Jpv5k1XAw6uKseOicwCFkWa63E5aWSiGxmtmHs 1938
colossalai/utils/memory.py sha256=aSJ3gGdX0r40L7fdjcQr2MOCkdWjFYEKOH4Xmg2h5_A 2470
colossalai/utils/timer.py sha256=UUaev4bkJHUsrXiL7wPjglPWrrMBtP5ebbxnqnCZQiA 4321
colossalai/utils/model/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colossalai/utils/model/utils.py sha256=dEpaGkDX4VncR7YTUNQphTlPBR_kZEDEkhfzIQVXciM 3905
colossalai/utils/multi_tensor_apply/__init__.py sha256=7KS_pngMOARZTO8UGSQY46vzaILuXjaamzyTQd9Q3x4 101
colossalai/utils/multi_tensor_apply/multi_tensor_apply.py sha256=9wWV2zwUwQyiiLtA2W4puGfFEo2bkG-XIJ-Ffm5NaSk 1142
colossalai/utils/rank_recorder/__init__.py sha256=m8OU7nLuj7p1O9sPM28Zm4jr0HUwv1ZHSr1zDrW2qKA 90
colossalai/utils/rank_recorder/rank_recorder.py sha256=YsNvIJeMmOWyA-qv6ELnokh9OJHgbHXdZfZtQCSs8M8 5453
colossalai/utils/tensor_detector/__init__.py sha256=kEFcs8ZKg9AVwesh9wU8NGaqPGvO_jGNooE9JKgjgUs 44
colossalai/utils/tensor_detector/tensor_detector.py sha256=TO6r713pxkZ5kVSVZYDHvG47g05rECb_ndhI6ufm-pw 8392
colossalai/zero/__init__.py sha256=vECJp4JIJWBHqkS9N72f2yVG1ITH6j29ZoydA7rJgso 390
colossalai/zero/wrapper.py sha256=AVEh5dySenvCyCaMTKzijyDW1XTGnRlMDiCIV_kWM10 4898
colossalai/zero/gemini/__init__.py sha256=voILUoE5PVnkevrR42DMZaZRpM_2HI9gID9XbRhq_4w 490
colossalai/zero/gemini/gemini_ddp.py sha256=UtLyNqUMiN6Iw8skDBfl4DePuBlrnQCQ7Y8UTLFtmX8 41791
colossalai/zero/gemini/gemini_hook.py sha256=ali9aoFS0QXCWRUDfqpk3rMi6_fZB4QgtK1tbYuMtgo 2458
colossalai/zero/gemini/gemini_mgr.py sha256=cWhCURLzaycJiUb-JtUpCXI69OGB26vcqx9KwB1Mayo 6243
colossalai/zero/gemini/gemini_optimizer.py sha256=Ugia2daMJzOJmT4euLRAG5cBZcjfDEY8DVA0J0zKLLU 36398
colossalai/zero/gemini/placement_policy.py sha256=6yHqD969EpbyJHdF8XtjjHUDo7OehIESMuPR5cZ56aE 9569
colossalai/zero/gemini/utils.py sha256=nhlYGyB69ckIlhYlGFG9lHSD33rIddspuFpHj14OkfY 4213
colossalai/zero/gemini/chunk/__init__.py sha256=v5BZxP230JuzOFNNXC4l4VYWcTJqVkBuh7zLAqlAJuw 342
colossalai/zero/gemini/chunk/chunk.py sha256=qY0RSguaeRuZ9DbedTKeRy8tcprCVYFJJ2oeEeojRa4 25030
colossalai/zero/gemini/chunk/manager.py sha256=pYaDHQN5z6qQShRYjq44t07Npl-AjLyhPt3p_LKm6uA 11791
colossalai/zero/gemini/chunk/search_utils.py sha256=1UF9_wwXTyQYbII7PfOp-9ekTKxdqbuMQ-oOH9qBgEA 6291
colossalai/zero/gemini/chunk/utils.py sha256=eEJCi6QmjRMeRr1B7Ez-KNSWSDdSQxvXue72uxegpZk 1428
colossalai/zero/gemini/memory_tracer/__init__.py sha256=nNoPR3wBmJKxINUH0L_1gX2JrxcG4OoLlRAG4QfG1Ek 511
colossalai/zero/gemini/memory_tracer/chunk_memstats_collector.py sha256=NZQ-3msgcgIvqpTtwS_C330HAtCbZ2sMcCWYd-udZIs 1281
colossalai/zero/gemini/memory_tracer/memory_monitor.py sha256=tEXlZc3oKXVp9P_KadY0V4Y0VJMJCjfq12Z9uQjLAbc 4063
colossalai/zero/gemini/memory_tracer/memory_stats.py sha256=t1lzpa4p1QWZpWImsgSasbNvRsQdSS4RE4W5wGBnKSw 4101
colossalai/zero/gemini/memory_tracer/memstats_collector.py sha256=P88Y0vjaj67MADJ6oUexqrhxgfPXvaiVXtX4Q6oBZFM 3591
colossalai/zero/gemini/memory_tracer/param_runtime_order.py sha256=YRfrD-FQxXQ7spAsYlMrEeY77Sif-RUR96isuvNddeg 858
colossalai/zero/gemini/memory_tracer/runtime_mem_tracer.py sha256=ZzPYWxFxhv2UrW4d6EZmVkyW0hxRgTlk7laWPf5ibVo 3716
colossalai/zero/gemini/memory_tracer/static_memstats_collector.py sha256=zFaVC5u-2pj0TBT9FJbxBZR61v2qdD_PxXT_yLgNn0A 4126
colossalai/zero/gemini/memory_tracer/utils.py sha256=3b-SMjp3saOgz3rwPbehkefwxOdec24TK2KtIFVNBtc 1795
colossalai/zero/low_level/__init__.py sha256=fDqSoLurzDHiyap5GqrOjDxaMRYziS0Ly3aNSrmEbMc 88
colossalai/zero/low_level/_utils.py sha256=coLDjyW2s6Uh80FlYqoFSuSJw8oe7rplmt3HQjVNJqQ 7282
colossalai/zero/low_level/low_level_optim.py sha256=4Mg28vkLPSFml7KycnlM56d_8mZGrEXw5peEsiogLO8 42465
colossalai/zero/low_level/bookkeeping/__init__.py sha256=YoBAZ1ou3lLtAmrUBPEetd9XZP6eY45V8hdBK9gF_eg 242
colossalai/zero/low_level/bookkeeping/base_store.py sha256=yBjaFNizZp3UPg8PRpotc0Atyj5Id9gCaMLWABNAZu0 409
colossalai/zero/low_level/bookkeeping/bucket_store.py sha256=qHtnpdgvoYHSD4eNPZ2vLcOChZ1zNhocUkH21Z2ZBuQ 4487
colossalai/zero/low_level/bookkeeping/gradient_store.py sha256=D4OLY-p98qXftcLvBbEGbJYH9jEagxAsMfbHnPxisk4 4136
colossalai/zero/low_level/bookkeeping/parameter_store.py sha256=ux9Lmt0BHTOVNzJRY-9XL8Q1GK1W7GvL9F7KVWrKypA 1502
colossalai/zero/low_level/bookkeeping/tensor_bucket.py sha256=TmdWQlksGcxt5tqIILmbNIh0VkNyjdXLk6vy4HFJgCY 1494
extensions/cpu_adam/__init__.py sha256=hUCxmVFMCigXdFoRuCUYEvFUtun_m4dR1A1AeAbC2yc 151
extensions/cpu_adam/cpu_adam_arm.py sha256=rAAv2eaVjIT6Z_UYOmtyZQ62ayehqla90Tqdf554T3k 1043
extensions/cpu_adam/cpu_adam_x86.py sha256=T9AlMnEU3bcfPA4NlJ7VgGTb_g7FtXvn192OHnNIRoo 1660
extensions/csrc/__init__.py sha256=UzC58VkROMaJgM1Vs_Upx88Ta4QI_GJZYR_lpdMaySo 350
extensions/csrc/scaled_softmax.py sha256=MUHkpayTNpJSPfLjtmSzOkMNQOUaN_t8YI1tWVymz88 6588
extensions/flash_attention/__init__.py sha256=xjTAEGepotS8ezBSU_DDCcRlm7x0hKnJx9TEmVFuy5M 527
extensions/flash_attention/flash_attention_dao_cuda.py sha256=VgXoWL7w0ReGds5WThnOf6T2YLkBRyPDy9FzGe_93CE 3502
extensions/flash_attention/flash_attention_npu.py sha256=1QeLKo919POHDwhNLgKrn14o_toJ-QnyZwE0jCBmEmA 2360
extensions/flash_attention/flash_attention_xformers_cuda.py sha256=w83ED0TebXXmTiDuJc0f6SlZR6MH8T4WRRAPX7ozbec 3622
extensions/layernorm/__init__.py sha256=fUNamrXwENT5vcBLrPGPkjBcKA2rzu2SEDxO5Cyq0d0 88
extensions/layernorm/layernorm_cuda.py sha256=iEWWZ-Q5avyCqy8KwZeEhXIh87tbJm2mVDzsNBRaDYg 822
extensions/moe/__init__.py sha256=XEM9bhqREZWBCag8thn6nFdLPuQuwW0pxmcTjx9gRHg 70
extensions/moe/moe_cuda.py sha256=OocQN4_roGwLJqSEdbrDqfKWG3NInYtSi9EyD_BvDFg 959
extensions/optimizer/__init__.py sha256=H4kcmlwol0VR6dM1Vfq4U0qWR9ZMhQVX2x1K1WS2XLo 104
extensions/optimizer/fused_optimizer_cuda.py sha256=x2lHMOkWsM1kXFUodIP3kjgieiSlzFkkmMisAA7_MXk 1104
extensions/softmax/__init__.py sha256=sTv-jI2bukCzTczJC0EbAzN7Ag3oqqS0lYQnjQfAm6c 271
extensions/softmax/scaled_masked_softmax_cuda.py sha256=A37wtyB_MwfvZcAfHeTXMrVlbPo93lxXOaqqa2CQN8U 1027
extensions/softmax/scaled_upper_triangle_masked_softmax_cuda.py sha256=zrvg39Gu_TUssvv03bxJx5vAFoty2WOO_KQOlfrRkhY 1098
tests/kit/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/kit/model_zoo/__init__.py sha256=B5mB4q50-kgjVE2QrT5WQLrEGWKPBRbnZCSIZdFJT7Y 1069
tests/kit/model_zoo/executor.py sha256=MsNCdA0WKdZRtycEeOVfByuEYI7eeOpsU-DfqvBLbx8 1499
tests/kit/model_zoo/registry.py sha256=2-ETXayRcb0F1mR-nAddxFfG4V_WCtZBzQ8Ys-P92YE 3413
tests/kit/model_zoo/custom/__init__.py sha256=GYek4qzE2XTcq2OPHtpNj456SJJmsVtIO7A2mQBDBpU 129
tests/kit/model_zoo/custom/base.py sha256=EKm898eaiAawuX6N5-fxEEI-ifXNhD4ee1yiqQNGz4E 832
tests/kit/model_zoo/custom/hanging_param_model.py sha256=YeMjkhEOWZw3Nl1JwCc_rnIm8bA85WC_xJBB8QRE940 1140
tests/kit/model_zoo/custom/nested_model.py sha256=CtT7E9TfFg3VIL_EFKsjfeSOksgd8SiJXgZHvV3dpBE 1206
tests/kit/model_zoo/custom/repeated_computed_layers.py sha256=1lZzEWBIV-jEU6OubxgvjNrx9vmuuPwSFqmxrLkOpqQ 1206
tests/kit/model_zoo/custom/simple_net.py sha256=kQOEgU9adMOhmi596-KJL1ro49ejG87707JzBJebbH0 1259
tests/kit/model_zoo/diffusers/__init__.py sha256=hXUWEGV4SnsRGaArzZZ3j_IEySAOsNRwI01QmTO3UO8 25
tests/kit/model_zoo/diffusers/diffusers.py sha256=VeNT8OjFeUvFCWNGgPSGhT67cCYQ75Qrc_PN0UeLj8c 2534
tests/kit/model_zoo/timm/__init__.py sha256=5Jak_jSF1txkZkezw-vrkEJk1Q1z-vVsxyeoxmNKesM 20
tests/kit/model_zoo/timm/timm.py sha256=9zQ0KE0tXrVhR9dSoLWvhurDdfkYxiKCxXTblyvdEBg 5975
tests/kit/model_zoo/torchaudio/__init__.py sha256=jD2o1vG3f-f7Z2gkQ3ZkSdP9MlVhvgQvbw3CvEYcOSE 26
tests/kit/model_zoo/torchaudio/torchaudio.py sha256=zX6o_SHLzjPxRkW_XZ8tqRNF7AueodJOLpBJIvxL4Z8 4614
tests/kit/model_zoo/torchrec/__init__.py sha256=wbj91pPhPAsSphi8QO1pcm2fqvaCtEkz_vgpKHcqjHE 24
tests/kit/model_zoo/torchrec/torchrec.py sha256=uPu2zk9Cqi5njb8HVW1jtLNy9rUvqYiWO2LBnTOWv5k 3899
tests/kit/model_zoo/torchvision/__init__.py sha256=d4M6fBJrxMRAC6l6hHPwML1amm6-6oil__d8yJxop9k 27
tests/kit/model_zoo/torchvision/torchvision.py sha256=PVthg5JdJ9ifpQmznsG57EiGj79BMN6dLYBarxlcbik 4970
tests/kit/model_zoo/transformers/__init__.py sha256=UDhiDQPTD3VBiNL_aDLchxy1nBvf1HAs4-43Wz1xlCc 408
tests/kit/model_zoo/transformers/albert.py sha256=pzAKDz_mmB92IomtAJB19Beat2aHcKl7AoN8dQM07yY 3768
tests/kit/model_zoo/transformers/bert.py sha256=dyR4VYSOTv5ZG8OsytgDFlGjsQ2Yxqyw5VGBE1X7FKE 12561
tests/kit/model_zoo/transformers/blip2.py sha256=xcAQiYjoxTh4K7mXyPmikVtwWvtypVDF6xXKWiRlE40 2289
tests/kit/model_zoo/transformers/bloom.py sha256=R9Hb9g9ggF-OfO1RTPJ_oWTYnlunlBHuKWdvgKdHWeY 4487
tests/kit/model_zoo/transformers/chatglm2.py sha256=tFa61otPYUUQom7fJiiRh4ZZfvFKUtTq6jPw-DVPaWw 2313
tests/kit/model_zoo/transformers/falcon.py sha256=WqxucvqD8mNSZ-w60V8pptQNbk4ojfuIQKY1YTG20Mo 4251
tests/kit/model_zoo/transformers/gpt.py sha256=fgpsHPuzVlHBQ8cpiit9tp8wyb9Rz6BpSfsgk9aLV4A 5466
tests/kit/model_zoo/transformers/gptj.py sha256=ntZaIQS5m2LFl1Idnlfjm6-yug7qFzwpL8J32CLO9Mo 3597
tests/kit/model_zoo/transformers/llama.py sha256=cfCxcecZlrdeQTPRoDWLG1u40_FA4AJwRge8etaAqKI 3097
tests/kit/model_zoo/transformers/mistral.py sha256=8duuerQ-ZE7-due4pva7zvyw-J6if4ftYi23muDg6uk 2754
tests/kit/model_zoo/transformers/opt.py sha256=-zdsc17ynlb6zyhiyQkDC8DhXEOVwqb8PK0z5q15vfE 3063
tests/kit/model_zoo/transformers/sam.py sha256=mskCzWam_9OCoXzAGntjNrXhkWJHMt-5ifpTnkjeAKo 1833
tests/kit/model_zoo/transformers/t5.py sha256=CfIz1o_q0pZNqcua1p9BkVu_M09vEPDzv-nMc8TTZOA 3061
tests/kit/model_zoo/transformers/vit.py sha256=Gg6rlAwn1OlGkgwsHX_MJ1bavFu9Mziw9XMxcuONc-w 2214
tests/kit/model_zoo/transformers/whisper.py sha256=qTUp1Tc7a8haUDGw7hqyVNMWjusljaUNcDuw1QZ90Zc 3717
tests/test_analyzer/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_analyzer/test_fx/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_analyzer/test_fx/test_bias_addition.py sha256=Qok7jAihTgHtOXjy6tGqo2LHVKGdW_Xu8QIAVhpb_do 3684
tests/test_analyzer/test_fx/test_mod_dir.py sha256=6sl33JtDqHqR7gI7-L-t-em4FXREdUlGr8BM6XC7lFQ 2139
tests/test_analyzer/test_fx/test_nested_ckpt.py sha256=KBQ6HDpcQZKHBgKQmVTbMcnyyT0rG5zqIJs7T4Rdw4M 1750
tests/test_analyzer/test_fx/test_shape_prop.py sha256=e0LyA5AlVzVSiwuhJlCAJYhHcWTN1xjibOU0rZj57f0 2127
tests/test_analyzer/test_fx/test_symbolic_profile.py sha256=Ayaww3qOzKqeYb0MNwgqvJ8aKvx4V3QuBUohMXYndY8 1718
tests/test_analyzer/test_fx/zoo.py sha256=aSQP5Inij_pWIrFpXeqvv8fX6Qrgkih6vPo8-oVzIJo 1271
tests/test_analyzer/test_subclasses/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_analyzer/test_subclasses/test_aten.py sha256=1hScIqVmq69Q4NXgLCIfRJHivOrsW0g1PlbdpeD8XHg 3714
tests/test_analyzer/test_subclasses/test_flop_tensor.py sha256=s4XBiBV3Gz0ddKZdbZZbkaoWeM1_afT5XTXGcvHKkdo 1893
tests/test_analyzer/test_subclasses/test_meta_mode.py sha256=7fcxPTuCLiqg8ecCWF4N1IY1jTJP0REkcglwc_GVgV0 1518
tests/test_auto_parallel/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_auto_parallel/test_pass/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_auto_parallel/test_pass/test_node_converting_pass.py sha256=282_TBaCfPcoFR0rFXQM8uOHXfF4lTPDgmxDaa_BYZA 1773
tests/test_auto_parallel/test_pass/test_size_value_converting_pass.py sha256=hARpI2nys8wrP1HoI3sFJsNmlFB8Ex3V8sY6gkrtSQ0 2250
tests/test_auto_parallel/test_tensor_shard/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_auto_parallel/test_tensor_shard/test_bias_addition_forward.py sha256=S8JJo4UcNQVgvaT7zTmUtbFTvkWfbGmENHN5ocDL2yU 2724
tests/test_auto_parallel/test_tensor_shard/test_broadcast.py sha256=1IeE7Sm8ysLqNu03Tb2kpWNdElBcSX1SS2Pw8ngf6EQ 2003
tests/test_auto_parallel/test_tensor_shard/test_checkpoint.py sha256=_U8TPRDXmFsk-32VVKXScUo7d2lCFpVew8GHgnb0eaM 2457
tests/test_auto_parallel/test_tensor_shard/test_compatibility_with_ddp.py sha256=a0W8ldAkSTx3ljqWDgFpRaDMng6bhyXI7T6DBskhuA0 3394
tests/test_auto_parallel/test_tensor_shard/test_compatibility_with_gemini.py sha256=aaRixXoSa1eT42KHrF7YqFbVQa3DNjBLBgcip45Me7c 3744
tests/test_auto_parallel/test_tensor_shard/test_find_repeat_block.py sha256=iV5NQrrLcADaMvbSCF2vQXu2mnGEDahHEd5jVt2SS48 3674
tests/test_auto_parallel/test_tensor_shard/test_liveness_analysis.py sha256=b8vBX2Sw59bN4Q_NtJemlYW7du14ILHBjrooP8jfaeE 2052
tests/test_auto_parallel/test_tensor_shard/test_solver_with_resnet_v2.py sha256=RWyQyQZI6tYl97mmCzMci8uwf03dL4AruOD-AEd_dAs 5283
tests/test_auto_parallel/test_tensor_shard/test_gpt/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_auto_parallel/test_tensor_shard/test_gpt/gpt_modules.py sha256=S7tJJK__3zHn9JbNrYXXEqOsbsAaCiHc94K_iuUGPHI 10992
tests/test_auto_parallel/test_tensor_shard/test_gpt/test_runtime_with_gpt_modules.py sha256=18tVeBNZdBHaR5WK1hdZw5FRvJKQZSgIvkrhNgpXmuo 7740
tests/test_auto_parallel/test_tensor_shard/test_gpt/test_solver_with_gpt_module.py sha256=fbbscqrDhStb2NTZEUnXUNNL3p1zTx3Ajk56EET5Ivw 3818
tests/test_auto_parallel/test_tensor_shard/test_node_handler/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_addbmm_handler.py sha256=Y4SJObsHnKfm5fDuKS8_ImA16TeqsrTd05AkkpMeMFA 11142
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_addmm_handler.py sha256=MdiM1VhR7eO6lY2F0p3SE8aD9ayKKWZCA7iGWd02UxI 7532
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_batch_norm_handler.py sha256=w0o8G6QlXhSRPnE2XvYZ50qwbROJIMAj0DEPqevX3p4 4712
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_bias_linear_function_node.py sha256=k-l-qJV70tw--Q9IFwtGct-7qBWeQNq6obMitktEXeE 6559
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_bias_linear_module_node.py sha256=99m3iywo5rfqGfapbjc4a_EDfM0-BMCD3nWKEgHpEFo 6045
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_binary_elementwise_handler.py sha256=O63py4ZTJblWdjozo1Ab-Lzm5dvcRVV7jgVqSus5Tn8 10730
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_bmm_handler.py sha256=j4F_cXv7ipPCJ8EY-vMBC0S-4W_TI9602hAZcty72kk 8739
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_conv_handler.py sha256=-rcbuBqh3ANfIbTssA-MnXEy1TTMXCL4unOoUbKgoW0 12596
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_default_reshape_handler.py sha256=-OFpgrXD4H9GpQD5Lo9tEss6okQ2mTwVRTO_hJJdDSU 3595
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_embedding_handler.py sha256=CTEchMjNyInuZrRuGb11QUU9z_RLcIfGFE7Q2n5G6Gk 11448
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_getattr_handler.py sha256=H0k3mZj_xbzF3DhOa7vfHDzIJcYXvoBW-mI8xXrEoc8 2942
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_getitem_handler.py sha256=3dLEwnW_aIT6_N-Dg9ff_x_pZqmOYdldY--ztQumnoA 7719
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_layer_norm_handler.py sha256=r24NUyUa1GQyfGAjCGFBL5XTVM_BiZYKK6tI0jT3Z-k 4242
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_linear_handler.py sha256=8ZR8vKtGZsJD62zWl_N5uTZNGFlX5yfywnSmt4vrTuw 12847
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_matmul_handler.py sha256=k2VpPmBNbCCTNkFe7enAF78tc0c9c12ElNDe9Xccy1c 8338
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_norm_pooling_handler.py sha256=sf0UqDNXh7dWjFUbRKGjm3cAJsJBVwiFkGcAHmEGLAQ 2552
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_output_handler.py sha256=tnHhhx2qRbgEQDYoi0BsQp2_K0JxbzBupY5pymBsPQE 2556
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_permute_and_transpose_handler.py sha256=2gU9kmvmWinBsVRXblD_FG9LuUjAsGGYSOxszULtY0A 18275
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_placeholder_handler.py sha256=JqwV37wFLRjVSGjpXwDNH8-7YHHNUFtwZdXjs2aHHHE 3056
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_shard_option.py sha256=duQESZ_yDAzFr4yljaIYT_9XtW49FzMWGQPgxsLuRX0 4168
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_softmax_handler.py sha256=3IDtKjoGRuSOfGW_fjfvrqczQKcfcMgkVHpt4Tk7hsQ 8709
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_split_handler.py sha256=r_pbu9FbkjfK39dRcDE4GmnUO-Vh9oP5L1_CG8ojxXw 12227
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_sum_handler.py sha256=Uy3cIsnWvMilHKmDuG99_SNoyYnuOCH4oQSaAC6eIV4 12148
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_tensor_constructor.py sha256=3bMAa6zSa_lOBrMZo4JfBJUva6Y5Y9lNW92ZfO3Ty_g 2781
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_unary_element_wise_handler.py sha256=PoWNORItFHiPwszSiLwstID3oxg1pTb41u1fl2gruHE 3625
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_view_handler.py sha256=Z5xbHHUdV9Lkdn4WhmVOeYkP7lcNaKbu10FjJr0wyRg 13858
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_where_handler.py sha256=4C-yoR7SurJ5HzkTn8Wc7KyosY1Zi9JIwZY0DBdon-w 3556
tests/test_auto_parallel/test_tensor_shard/test_node_handler/utils.py sha256=XhBAGyY5PC6Qh5IkQTi0ih3qgAFzAi66rBZlzgXmVyE 8505
tests/test_shardformer/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_shardformer/test_shard_utils.py sha256=1eRKCf8Zz1vZQmZfybnOGXGmRHk4OKerGEu23AfyVcI 833
tests/test_shardformer/test_with_torch_ddp.py sha256=_YI31WzEy97psyGudNOnGyK0zpx3afW08FLK2LnCYoY 2634
tests/test_shardformer/test_model/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_shardformer/test_model/_utils.py sha256=xX0b2NMAr2Dc9T0CopF3UM2dPcqmS9QpUNSHpbWXlXw 12151
tests/test_shardformer/test_model/test_shard_bert.py sha256=cjnUN1c9Wp9SQ4rdRX3byMQky_JAG8HL1J7Uk4swmGQ 7653
tests/test_shardformer/test_model/test_shard_blip2.py sha256=P8rT0-L5QLjsPt-zv40Bgrjh_vfoUdLhZ4QK1bkdm2c 2888
tests/test_shardformer/test_model/test_shard_bloom.py sha256=c7tmNItEb8npkWnsH_cYh_yeY8-5tT2oHJ7TNNzMaHQ 7035
tests/test_shardformer/test_model/test_shard_chatglm2.py sha256=qkR9Ui6H146Txt3VE2q2rytrJR_eGY5lIzB_ClqvOfs 7154
tests/test_shardformer/test_model/test_shard_falcon.py sha256=Tkv32VZzwGxcPLZvSzJHMIAzkWl_vjfLPHG8pt6DPxY 6670
tests/test_shardformer/test_model/test_shard_gpt2.py sha256=8NivjEbFHixBnSv-u-c9My8oCAaqxSJaMLt93CDf6Hs 7430
tests/test_shardformer/test_model/test_shard_gptj.py sha256=gLkk07N3lFZgFmIx_TTV0owlyRdQP4P0iJrqjY-dMQo 7209
tests/test_shardformer/test_model/test_shard_llama.py sha256=_tP58CVcvz7I5Oa7xV55ay67FBGnRblPVD5NWJd51TM 7417
tests/test_shardformer/test_model/test_shard_mistral.py sha256=YTSqpwYact8u3X0qCKgvphsa0exet1ND2xyFCVOIRg8 5225
tests/test_shardformer/test_model/test_shard_opt.py sha256=E1ry8p4XGiBTKjgJX3IIHaAXCLWAd971L8Vg5w8-cTY 6765
tests/test_shardformer/test_model/test_shard_sam.py sha256=r0L8Uw5EICBxK3PPjcOjoKWUQEZldlMOy8vnw8aPvXI 2568
tests/test_shardformer/test_model/test_shard_t5.py sha256=SHQ7yEf6pWYjjRFDgJIVMRQhHwufZyBfEBocDdseSh0 6953
tests/test_shardformer/test_model/test_shard_vit.py sha256=9WzFtIC-yTugpw3ge7scWWwD6_JkMki4nL1qIoudu3w 6550
tests/test_shardformer/test_model/test_shard_whisper.py sha256=eDOCaAG3MOf16uHQ3mxni8AGrJeOdpGuAr8_3lphF3c 7314
colossalai_nightly-2024.2.19.dist-info/LICENSE sha256=KGm5Fom6FNIN5E0VOXf1P1ja0OvlgsUhjDPzbliVKcY 29471
colossalai_nightly-2024.2.19.dist-info/METADATA sha256=8z87BM605H0hnwhXVRLrwGsYzfRZm6-k1EAWytqz53g 30105
colossalai_nightly-2024.2.19.dist-info/WHEEL sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM 92
colossalai_nightly-2024.2.19.dist-info/entry_points.txt sha256=OVk4YQrW80Oya9BG9MWwMvnoI5vAyiyjhtfPFSl9KeY 69
colossalai_nightly-2024.2.19.dist-info/top_level.txt sha256=37ajyIWgPmCTEoil40kqXqjQTZb38v-jK7iUpxobepY 28
colossalai_nightly-2024.2.19.dist-info/RECORD — —

top_level.txt

colossalai
extensions
tests

entry_points.txt

[console_scripts]
colossalai = colossalai.cli:cli