Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/tilelang.git" did not exist on "b14f201ef5d5db28accd001dccbaf799253a2741"
Unverified Commit cbe34c55 authored by Rocky Duan's avatar Rocky Duan Committed by GitHub
Browse files

Fix ColoTensorSpec for py11 (#5440)

parent a7790a92
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
colossalai/legacy/tensor/tensor_spec.py
View file @ cbe34c55
from dataclasses import dataclass from dataclasses import dataclass, field
from typing import Optional from typing import Optional
from colossalai.legacy.tensor.distspec import DistPlacementPattern, _DistSpec from colossalai.legacy.tensor.distspec import DistPlacementPattern, _DistSpec
...@@ -17,5 +17,5 @@ class ColoTensorSpec: ...@@ -17,5 +17,5 @@ class ColoTensorSpec:
""" """
pg: ProcessGroup pg: ProcessGroup
dist_attr: Optional[_DistSpec] = _DistSpec(DistPlacementPattern.REPLICATE) dist_attr: Optional[_DistSpec] = field(default_factory=lambda: _DistSpec(DistPlacementPattern.REPLICATE))
compute_attr: Optional[ComputeSpec] = None compute_attr: Optional[ComputeSpec] = None
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment