github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-18 07:31:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

[devops] update torch version of CI (#3725)

Browse Source 
* [test] fix flop tensor test

* [test] fix autochunk test

* [test] fix lazyinit test

* [devops] update torch version of CI

* [devops] enable testmon

* [devops] fix ci

* [devops] fix ci

* [test] fix checkpoint io test

* [test] fix cluster test

* [test] fix timm test

* [devops] fix ci

* [devops] fix ci

* [devops] fix ci

* [devops] fix ci

* [devops] force sync to test ci

* [test] skip fsdp test
This commit is contained in:
Hongxin Liu
2023-05-15 17:20:56 +08:00
committed by GitHub
parent b37797ed3d
commit afb239bbf8
17 changed files with 74 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_fx/test_tracer/test_timm_model/test_timm_model.py
View File

@@ -43,6 +43,12 @@ def trace_and_compare(model_cls, data, output_transform_fn, meta_args=None):
f'{model.__class__.__name__} has inconsistent outputs, {fx_output_val} vs {non_fx_output_val}'
# FIXME(ver217): timm/models/convit.py:71: in forward
# if self.rel_indices is None or self.rel_indices.shape[1] != N:
# torch/fx/proxy.py:284: in __bool__
# return self.tracer.to_bool(self)
# torch.fx.proxy.TraceError: symbolically traced variables cannot be used as inputs to control flow
@pytest.mark.skip("convit is not supported yet")
@pytest.mark.skipif(version.parse(torch.__version__) < version.parse('1.12.0'), reason='torch version < 12')
@clear_cache_before_run()
def test_timm_models():