custom
code
sovereign-compute
nvidia-stack / mfma-core /asic /scripts /mfma_core_layout.py
SNAPKITTYWEST's picture
chore: push from SNAPKITTYWEST local build
e92f76f verified
Raw
History Blame Contribute Delete
825 Bytes
#!/usr/bin/env python3
# mfma_core_layout.py — GDSFactory Layout Generation for MFMA Core
import gdsfactory as gf
@gf.cell
def mfma_tile_layout():
c = gf.Component("mfma_tile_gdsii")
# Core systolic array footprint (16x16x16 FP16 MAC units)
core = c << gf.components.rectangle(size=(128.0, 128.0), layer=(1, 0))
core.name = "mfma_systolic_core"
# Metal 1/Metal 2 power distribution network (PDN) rings
vdd_ring = c << gf.components.rectangle(size=(132.0, 132.0), layer=(3, 0))
vdd_ring.center = core.center
gnd_ring = c << gf.components.rectangle(size=(136.0, 136.0), layer=(4, 0))
gnd_ring.center = core.center
return c
if __name__ == "__main__":
c = mfma_tile_layout()
c.write_gds("mfma_core.gdsii")
print("Successfully generated GDSII stream: mfma_core.gdsii")