From 7f0f6c860d48edaf9db43ddf010e80f0bac75447 Mon Sep 17 00:00:00 2001 From: Ivan Basov Date: Mon, 30 Mar 2026 11:54:58 -0700 Subject: [PATCH 1/3] fix(ci): disable torch.compile in orientation training to prevent segfault torch.compile=on combined with DataLoader spawn workers during LER validation causes a segfault (20 leaked semaphores, core dumped). Set PREDECODER_TORCH_COMPILE=0 for the Train all orientations step. Co-Authored-By: Claude Sonnet 4.6 --- .github/workflows/long-running-tests.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/long-running-tests.yml b/.github/workflows/long-running-tests.yml index f536780..3c9e268 100644 --- a/.github/workflows/long-running-tests.yml +++ b/.github/workflows/long-running-tests.yml @@ -184,6 +184,7 @@ jobs: PREDECODER_VAL_SAMPLES: "4096" PREDECODER_TEST_SAMPLES: "4096" PREDECODER_TRAIN_EPOCHS: "1" + PREDECODER_TORCH_COMPILE: "0" - name: Multi-orientation inference (O1–O4) with LER output check shell: bash From 9d3fa086d9768091054aabe95606fea3424002f6 Mon Sep 17 00:00:00 2001 From: Ivan Basov Date: Mon, 30 Mar 2026 11:57:04 -0700 Subject: [PATCH 2/3] Revert "fix(ci): disable torch.compile in orientation training to prevent segfault" This reverts commit 7f0f6c860d48edaf9db43ddf010e80f0bac75447. --- .github/workflows/long-running-tests.yml | 1 - 1 file changed, 1 deletion(-) diff --git a/.github/workflows/long-running-tests.yml b/.github/workflows/long-running-tests.yml index 3c9e268..f536780 100644 --- a/.github/workflows/long-running-tests.yml +++ b/.github/workflows/long-running-tests.yml @@ -184,7 +184,6 @@ jobs: PREDECODER_VAL_SAMPLES: "4096" PREDECODER_TEST_SAMPLES: "4096" PREDECODER_TRAIN_EPOCHS: "1" - PREDECODER_TORCH_COMPILE: "0" - name: Multi-orientation inference (O1–O4) with LER output check shell: bash From d7adb8079a6fa7e552c86ed0d7a4e44279b3e0b4 Mon Sep 17 00:00:00 2001 From: Ivan Basov Date: Mon, 27 Apr 2026 11:01:36 -0700 Subject: [PATCH 3/3] Add publication reference --- README.md | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index a0b398b..c6aba10 100644 --- a/README.md +++ b/README.md @@ -8,6 +8,17 @@ This repo implements a **pre-decoder** for surface-code memory experiments: The public release exposes a **single user-facing config** and a **single runner script**. +## Publication + +This implementation accompanies the paper: + +Christopher Chamberland, Jan Olle, Muyuan Li, Scott Thornton, and Igor Baratta, +"Fast and accurate AI-based pre-decoders for surface codes," +[arXiv:2604.12841](https://arxiv.org/abs/2604.12841), 2026. +[doi:10.48550/arXiv.2604.12841](https://doi.org/10.48550/arXiv.2604.12841) + +Please cite the paper if you use this repository in research or published work. + ### Quick start (train + inference) From the repo root: @@ -556,4 +567,4 @@ Presence of these headers is enforced automatically by the `spdx-header-check` C `.github/workflows/ci.yml`). Third-party open source components bundled with or required by this project are listed with their -respective copyright notices and license texts in [NOTICE](NOTICE). \ No newline at end of file +respective copyright notices and license texts in [NOTICE](NOTICE).