update atomic import (modular#242)

* update atomic import

* Update MAX to latest nightly

Author: dunnoconnor

pixi.toml

@@ -42,13 +42,17 @@ python = "==3.12"
 mojo = "<1.0.0" # includes `mojo-compiler`, lsp, debugger, formatter etc.
 <<<<<<< HEAD
 <<<<<<< HEAD
+<<<<<<< HEAD
 max = "==26.3.0.dev2026033105"
 =======
 max = "==26.3.0.dev2026032405"
 >>>>>>> 11c7cd4 (Mdoc/fixes (#235))
 =======
 max = "==26.3.0.dev2026033105"
 >>>>>>> d09bc3f (Update all implicit type casts to be explicit (#237))
+=======
+max = "==26.3.0.dev2026042005"
+>>>>>>> ff53970 (update atomic import (#242))
 bash = ">=5.2.21,<6"
 manim = ">=0.18.1,<0.19"
 mdbook = ">=0.4.48,<0.5"

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

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

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",

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
 

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

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