[mojo-lang] MOCO-3352: Update def to def raises for GPU puzzles (#230)

BEGIN_PUBLIC
[mojo-lang] MOCO-3352: Update def to def raises for GPU puzzles
END_PUBLIC

Author: fairlydurable

book/src/puzzle_04/intro.mojo

@@ -17,7 +17,7 @@ fn kernel[
     print(tensor)
 
 
-def main():
+def main() raises:
     ctx = DeviceContext()
 
     a = ctx.enqueue_create_buffer[dtype](HEIGHT * WIDTH)

problems/p01/p01.mojo

@@ -21,7 +21,7 @@ fn add_10(
 # ANCHOR_END: add_10
 
 
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out = ctx.enqueue_create_buffer[dtype](SIZE)
         out.enqueue_fill(0)

problems/p02/p02.mojo

@@ -22,7 +22,7 @@ fn add(
 # ANCHOR_END: add
 
 
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out = ctx.enqueue_create_buffer[dtype](SIZE)
         out.enqueue_fill(0)

problems/p03/p03.mojo

@@ -22,7 +22,7 @@ fn add_10_guard(
 # ANCHOR_END: add_10_guard
 
 
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out = ctx.enqueue_create_buffer[dtype](SIZE)
         out.enqueue_fill(0)

problems/p04/p04.mojo

@@ -23,7 +23,7 @@ fn add_10_2d(
 # ANCHOR_END: add_10_2d
 
 
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out = ctx.enqueue_create_buffer[dtype](SIZE * SIZE)
         out.enqueue_fill(0)

problems/p04/p04_layout_tensor.mojo

@@ -24,7 +24,7 @@ fn add_10_2d(
 # ANCHOR_END: add_10_2d_layout_tensor
 
 
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out_buf = ctx.enqueue_create_buffer[dtype](SIZE * SIZE)
         out_buf.enqueue_fill(0)

problems/p05/p05.mojo

@@ -22,7 +22,7 @@ fn broadcast_add(
 
 
 # ANCHOR_END: broadcast_add
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out = ctx.enqueue_create_buffer[dtype](SIZE * SIZE)
         out.enqueue_fill(0)

problems/p05/p05_layout_tensor.mojo

@@ -29,7 +29,7 @@ fn broadcast_add[
 
 
 # ANCHOR_END: broadcast_add_layout_tensor
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out_buf = ctx.enqueue_create_buffer[dtype](SIZE * SIZE)
         out_buf.enqueue_fill(0)

problems/p06/p06.mojo

@@ -22,7 +22,7 @@ fn add_10_blocks(
 # ANCHOR_END: add_10_blocks
 
 
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out = ctx.enqueue_create_buffer[dtype](SIZE)
         out.enqueue_fill(0)

problems/p07/p07.mojo

@@ -23,7 +23,7 @@ fn add_10_blocks_2d(
 # ANCHOR_END: add_10_blocks_2d
 
 
-def main():
+def main() raises:
     with DeviceContext() as ctx:
         out = ctx.enqueue_create_buffer[dtype](SIZE * SIZE)
         out.enqueue_fill(0)