Adding tt-metalium

recipes/tt-metalium/build.sh

@@ -0,0 +1,30 @@
+#!/bin/env/bash
+set -e
+
+# Needed so that sfpi can be invoked at build time to generate object files
+# sfpi links libmpc.so at runtime
+export LD_LIBRARY_PATH=$PREFIX/lib
+
+# Avoid overloading build machine processors and memory
+export NUM_PROCS=$((CPU_COUNT / 2))
+
+# Needed by python setup.py
+export TT_FROM_PRECOMPILED_DIR=$SRC_DIR
+
+cmake \
+  $CMAKE_ARGS \
+  -G Ninja \
+  -S $SRC_DIR \
+  -B $SRC_DIR/build \
+  -DCMAKE_BUILD_TYPE=Release \
+  -DCMAKE_SKIP_INSTALL_RPATH=ON
+
+cmake --build $SRC_DIR/build --parallel $NUM_PROCS
+
+cmake --install $SRC_DIR/build
+
+# Warning - HACK!
+mkdir -p $SRC_DIR/build/lib
+cp $PREFIX/lib/_ttnn.so $SRC_DIR/build/lib/
+
+pip install --no-deps $SRC_DIR

recipes/tt-metalium/conda_build_config.yaml

@@ -0,0 +1,4 @@
+# This package uses some RISCV defines from the sysroot "elf.h"
+# These defines are not available in the old sysroot
+c_stdlib_version:          # [linux]
+  - 2.34                   # [linux]

recipes/tt-metalium/meta.yaml

@@ -0,0 +1,73 @@
+{% set name = "tt-metalium" %}
+{% set version = "0.56.0" %}
+
+package:
+  name: {{ name|lower }}
+  version: {{ version }}
+
+source:
+  url: https://github.com/tenstorrent/tt-metal/releases/download/v{{ version }}-rc1/tt-metalium.tar.gz
+  sha256: 319f8c8f5b3ce98ef7db0c0e230cd37aca339b06ece98d923daadf4d86b27465
+
+# Skipping build for other python versions due to timeout in staging CI
+# Will enable more python versions after merge
+build:
+  number: 0
+  skip: true  # [not linux or py!=310]
+
+requirements:
+  build:
+    - {{ compiler('c') }}
+    - {{ compiler('cxx') }}
+    - {{ stdlib("c") }}
+    - cmake
+    - ninja
+    - git
+
+  host:
+    - python
+    - pip
+    - wheel
+    - setuptools
+    - numactl
+    - libhwloc
+    - mpc
+    - numpy
+    - loguru
+    - networkx
+    - pytorch
+    - python-graphviz
+
+  run:
+    - python
+    - numactl
+    - libhwloc
+    - libzlib
+    - mpfr
+    - gmp
+    - mpc
+    - numpy
+    - loguru
+    - networkx
+    - pytorch
+    - python-graphviz
+
+test:
+  imports:
+    - ttnn
+
+about:
+  home: https://github.com/tenstorrent/tt-metal
+  summary: 'Simple, fast, extensible runtime libraries for Tenstorrent Hardware'
+  description: |
+    TT-NN operator library, and TT-Metalium low level kernel programming model.
+  license: Apache-2.0
+  license_family: Apache
+  license_file: LICENSE
+  doc_url: https://docs.tenstorrent.com/tt-metalium/latest
+  dev_url: https://github.com/tenstorrent/tt-metal
+
+extra:
+  recipe-maintainers:
+    - blozano-tt
+    - afuller-TT