github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-01 17:17:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

[test] Fix/fix testcase (#5770)

Browse Source 
* [fix] branch for fix testcase;

* [fix] fix test_analyzer & test_auto_parallel;

* [fix] remove local change about moe;

* [fix] rm local change moe;
This commit is contained in:
duanjunwen
2024-06-03 15:26:01 +08:00
committed by GitHub
parent 3f2be80530
commit 1b76564e16
4 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/fx/codegen/activation_checkpoint_codegen.py
View File

@@ -859,7 +859,7 @@ if CODEGEN_AVAILABLE:
{wrap_stmts}
{prologue}
{code}"""
return PythonCode(fn_code, globals_)
return PythonCode(fn_code, globals_, {})
else: