ColossalAI
                                
                                 ColossalAI copied to clipboard
                                
                                    ColossalAI copied to clipboard
                            
                            
                            
                        [BUG]: the autoparallel compatibility test uses same input data for different data parallel rank
🐛 Describe the bug
The compatibility test for combination auto-parallel with Gemini used same input data for different data parallel rank. Therefore, we can't check whether the gradient hooks have been activated correctly or not.
Environment
No response