mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
8 lines
114 B
8 lines
114 B
2 years ago
|
from dataclasses import dataclass
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class MeshConfig:
|
||
|
|
TFLOPS: float = 1.9e12
|
||
|
|
BANDWIDTH = 1.2e9
|