code: def register(self, compute_pattern, pg: ProcessGroup):
      if not compute_pattern in self._allowed_patterns:
          if ComputePattern.TP1D == compute_pattern:
              self._set_TP1D(pg)
  
imports: from colossalai.legacy.tensor import ComputePattern, ProcessGroup, ShardSpec
  from .colo_module import ColoModule
label_desc: Use image node_id 2 for calling the ColoEmbedding obj's underlying member method code with example usage: obj.register(compute_pattern, pg) without return types