|
143 | 143 | "F6": {"gfx950": 131072}, |
144 | 144 | "F6F4": {"gfx950": 131072}, # Mixed precision F6 x F4 |
145 | 145 | "F8": dict.fromkeys(["gfx90a", "gfx940", "gfx941", "gfx942", "gfx950"], 32768), |
146 | | - "F16": dict.fromkeys(["gfx90a", "gfx940", "gfx941", "gfx942"], 16384) | dict.fromkeys(["gfx950"], 32768), |
| 146 | + "F16": dict.fromkeys(["gfx90a", "gfx940", "gfx941", "gfx942"], 16384) |
| 147 | + | dict.fromkeys(["gfx950"], 32768), |
147 | 148 | "F32": dict.fromkeys( |
148 | 149 | ["gfx908", "gfx90a", "gfx940", "gfx941", "gfx942", "gfx950"], 4096 |
149 | 150 | ), |
150 | 151 | "BF16": dict.fromkeys(["gfx940", "gfx941", "gfx942"], 16384) |
151 | | - | dict.fromkeys(["gfx90a"], 8192) | dict.fromkeys(["gfx950"], 32768), |
| 152 | + | dict.fromkeys(["gfx90a"], 8192) |
| 153 | + | dict.fromkeys(["gfx950"], 32768), |
152 | 154 | "I8": dict.fromkeys(["gfx940", "gfx941", "gfx942"], 32768) |
153 | | - | dict.fromkeys(["gfx90a"], 16384) | dict.fromkeys(["gfx950"], 65536), |
| 155 | + | dict.fromkeys(["gfx90a"], 16384) |
| 156 | + | dict.fromkeys(["gfx950"], 65536), |
154 | 157 | "F64": dict.fromkeys(["gfx90a", "gfx940", "gfx941", "gfx942", "gfx950"], 2048), |
155 | 158 | } |
156 | 159 |
|
|
0 commit comments