mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-23 10:30:03 +00:00
[refactor] remove gpc dependency in colotensor's _ops (#1189)
This commit is contained in:
@@ -52,5 +52,5 @@ class ColoModule(object):
|
||||
def get_param_names(self):
|
||||
return self._shard_params
|
||||
|
||||
def register(self, compute_pattern):
|
||||
def register(self, compute_pattern, pg):
|
||||
raise NotImplementedError
|
||||
|
Reference in New Issue
Block a user