update module 3

Author: srush

project/parallel_check.py

@@ -29,9 +29,9 @@
 # MM
 print("MATRIX MULTIPLY")
 out, a, b = (
-    minitorch.zeros((10, 10)),
-    minitorch.zeros((10, 20)),
-    minitorch.zeros((20, 10)),
+    minitorch.zeros((1, 10, 10)),
+    minitorch.zeros((1, 10, 20)),
+    minitorch.zeros((1, 20, 10)),
 )
 tmm = minitorch.fast_ops.tensor_matrix_multiply
 

tests/test_tensor_general.py

@@ -157,6 +157,16 @@ def test_sum_practice5():
         out = b2.sum(0)
         assert_close(s, out[0])
 
+    @pytest.mark.task3_3
+    def test_sum_practice_other_dims():
+        x = [[random.random() for i in range(32)] for j in range(16)]
+        b = minitorch.tensor(x)
+        s = b.sum(1)
+        b2 = minitorch.tensor(x, backend=shared["cuda"])
+        out = b2.sum(1)
+        for i in range(16):
+            assert_close(s[i, 0], out[i, 0])
+
     @pytest.mark.task3_4
     def test_mul_practice1():
         x = [[random.random() for i in range(2)] for j in range(2)]