You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/testing
Hongxin Liu 19e1a5cf16
[shardformer] update colo attention to support custom mask (#5510)
8 months ago
..
__init__.py [test] merge old components to test to model zoo (#4945) 1 year ago
comparison.py [shardformer] update colo attention to support custom mask (#5510) 8 months ago
pytest_wrapper.py [misc] update pre-commit and run all files (#4752) 1 year ago
random.py [misc] update pre-commit and run all files (#4752) 1 year ago
utils.py [npu] change device to accelerator api (#5239) 11 months ago