zml: initial release

Author: steeve

.bazelignore

@@ -0,0 +1 @@
+examples/.bazelignore

.bazelrc

@@ -0,0 +1 @@
+examples/.bazelrc

.bazelrc-examples

@@ -0,0 +1,2 @@
+common   --override_module=zml=../
+common   --registry=file://%workspace%/../third_party/

.bazelrc-zml

@@ -0,0 +1,2 @@
+common   --override_module=zml=./
+common   --registry=file://%workspace%/third_party/

.bazelversion

@@ -0,0 +1 @@
+examples/.bazelversion

.github/runs-on.yml

@@ -0,0 +1,11 @@
+images:
+  ubuntu24-amd64:
+    platform: "linux"
+    arch: "amd64"
+    ami: "ami-04a92520784b93e73"
+    preinstall: |
+      #!/bin/bash
+      wget https://github.com/jbdalido/bazel-remote/releases/download/test/bazel-remote -O /usr/local/bin/bzlcache
+      chmod +x /usr/local/bin/bzlcache
+      echo "Run bazel cache"
+      systemd-run --property=Restart=always /usr/local/bin/bzlcache --max_size=1000 --grpc_address 127.0.0.1:15501 --dir=/tmp/cache --s3.region eu-west-3 --s3.bucket runs-on-s3bucketcache-jycgicewbssh --s3.auth_method iam_role --s3.prefix cache --s3.endpoint=s3.eu-west-3.amazonaws.com --storage_mode uncompressed

.github/workflows/commit.yaml

@@ -0,0 +1,44 @@
+name: commit
+on:
+  push:
+    branches:
+    - master
+    tags:
+    - '*'
+  pull_request:
+
+env:
+  ZIG_VERSION: 0.13.0
+  BAZELISK_VERSION: 1.20.0
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: ${{ !contains(github.ref, 'master')}}
+
+jobs:
+  test:
+    runs-on: ["runs-on","runner=4cpu-linux-x64", "image=ubuntu24-amd64", "run-id=${{ github.run_id }}"]
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Setup Zig
+        uses: goto-bus-stop/setup-zig@v2
+        with:
+          version: ${{ env.ZIG_VERSION }}
+
+      - name: Format
+        run: zig fmt --check *.zig zml/ examples/ async/ mlir/ pjrt/
+
+      - name: No Autofix
+        run: bash -c 'grep "// autofix" zml/**/*.zig ; [ 1 -eq $? ]'
+
+      - name: Setup Bazel
+        run: |
+          wget https://github.com/bazelbuild/bazelisk/releases/download/v${{ env.BAZELISK_VERSION }}/bazelisk-linux-amd64 -O ~/bin/bazel
+          chmod +x ~/bin/bazel
+          
+      - name: Tests
+        run: |
+          export PATH=$PATH:~/bin
+          bazel test --remote_cache=grpc://127.0.0.1:15501 //zml:test --test_output=streamed

.gitignore

@@ -0,0 +1,30 @@
+# Bazel symlinks
+bazel-*
+
+# macOS
+.DS_Store
+
+# Local Checkouts
+/stablehlo/
+/llvm-project/
+/xla/
+
+# Just in case, Zig caches
+.zig-cache/
+zig-out/
+
+# Models
+*.safetensors
+*.pb
+*.trace.json
+*.pt
+*.bin
+
+# Data files
+*.wav
+*.ecdc
+.venv
+
+# Editor specific
+## Neovim
+.nvim.local.lua

.nvim.lua

@@ -0,0 +1 @@
+examples/.nvim.lua

.vscode

@@ -0,0 +1 @@
+examples/.vscode