From d1c4e6280548ec7ac6c7dfb705ee7455ae8da4e5 Mon Sep 17 00:00:00 2001 From: dunnoconnor Date: Mon, 20 Apr 2026 17:18:58 -0400 Subject: [PATCH 1/2] update atomic import --- problems/p22/op/layernorm_linear.mojo | 2 +- problems/p27/p27.mojo | 2 +- solutions/p10/p10.mojo | 1 - solutions/p22/op/layernorm_linear.mojo | 2 +- solutions/p27/p27.mojo | 2 +- 5 files changed, 4 insertions(+), 5 deletions(-) diff --git a/problems/p22/op/layernorm_linear.mojo b/problems/p22/op/layernorm_linear.mojo index dc659b12..2c3c558e 100644 --- a/problems/p22/op/layernorm_linear.mojo +++ b/problems/p22/op/layernorm_linear.mojo @@ -1,7 +1,7 @@ from std.math import sqrt from std.gpu import thread_idx, block_idx, block_dim, barrier from std.gpu.memory import AddressSpace, async_copy_wait_all -from std.os.atomic import Atomic +from std.atomic import Atomic from layout import TileTensor from layout.tile_layout import row_major, TensorLayout from layout.tile_tensor import stack_allocation diff --git a/problems/p27/p27.mojo b/problems/p27/p27.mojo index 6dbbd7de..55b9ef61 100644 --- a/problems/p27/p27.mojo +++ b/problems/p27/p27.mojo @@ -1,5 +1,5 @@ from std.gpu import thread_idx, block_idx, block_dim, grid_dim, barrier -from std.os.atomic import Atomic +from std.atomic import Atomic from std.gpu.primitives.warp import WARP_SIZE from std.gpu.primitives import block from std.gpu.host import DeviceContext diff --git a/solutions/p10/p10.mojo b/solutions/p10/p10.mojo index 7c25c4ed..450d2811 100644 --- a/solutions/p10/p10.mojo +++ b/solutions/p10/p10.mojo @@ -6,7 +6,6 @@ from layout.tile_layout import row_major from layout.tile_tensor import stack_allocation from std.testing import assert_equal from std.sys import argv -from std.os.atomic import Atomic # ANCHOR: shared_memory_race_solution diff --git a/solutions/p22/op/layernorm_linear.mojo b/solutions/p22/op/layernorm_linear.mojo index fe9fea3e..17715d17 100644 --- a/solutions/p22/op/layernorm_linear.mojo +++ b/solutions/p22/op/layernorm_linear.mojo @@ -1,7 +1,7 @@ from std.math import sqrt from std.gpu import thread_idx, block_idx, block_dim, barrier from std.gpu.memory import AddressSpace -from std.os.atomic import Atomic +from std.atomic import Atomic from layout import TileTensor from layout.tile_layout import row_major, TensorLayout from layout.tile_tensor import stack_allocation diff --git a/solutions/p27/p27.mojo b/solutions/p27/p27.mojo index 2f095125..f3c2fb35 100644 --- a/solutions/p27/p27.mojo +++ b/solutions/p27/p27.mojo @@ -1,5 +1,5 @@ from std.gpu import thread_idx, block_idx, block_dim, grid_dim, barrier -from std.os.atomic import Atomic +from std.atomic import Atomic from std.gpu.primitives.warp import WARP_SIZE from std.gpu.primitives import block from std.gpu.host import DeviceContext From cd5abeaf774aa3b9f4f953068a5a2fbae327c754 Mon Sep 17 00:00:00 2001 From: dunnoconnor Date: Tue, 21 Apr 2026 09:09:38 -0400 Subject: [PATCH 2/2] Update MAX to latest nightly --- pixi.toml | 2 +- pyproject.toml | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pixi.toml b/pixi.toml index 90cd7c4f..05c1c4d5 100644 --- a/pixi.toml +++ b/pixi.toml @@ -40,7 +40,7 @@ system-requirements = { macos = "15.0" } [dependencies] python = "==3.12" mojo = "<1.0.0" # includes `mojo-compiler`, lsp, debugger, formatter etc. -max = "==26.3.0.dev2026033105" +max = "==26.3.0.dev2026042005" bash = ">=5.2.21,<6" manim = ">=0.18.1,<0.19" mdbook = ">=0.4.48,<0.5" diff --git a/pyproject.toml b/pyproject.toml index 4e889c43..578da81d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,8 +4,8 @@ version = "1.0.0" readme = "README.md" requires-python = ">=3.10,<3.13" dependencies = [ - "mojo>=0.26.3.0.dev2026033105,<1.0.0", - "max==26.3.0.dev2026033105", + "mojo>=0.26.3.0.dev2026042005,<1.0.0", + "max==26.3.0.dev2026042005", "poethepoet>=0.34.0", "scipy>=1.15.3", "torch==2.7.1",