julia matmul: reorder loops & add inbounds (#2)

jariji · web-flow · commit aedad2b5fae9 · 2024-01-03T08:09:51.000-05:00
* julia matmul: reorder loops

* Add inbounds

* More inbounds

* Reorder matgen loop

* More concise
diff --git a/src/julia/matmul.jl b/src/julia/matmul.jl
@@ -1,19 +1,18 @@
 function matgen(n)
 	a = Array{Float64}(undef, n, n)
 	tmp = 1.0 / n / n
-	for i = 1:n, j = 1:n
+	@inbounds for j = 1:n, i = 1:n
 		a[i,j] = tmp * (i - j) * (i + j - 2)
 	end
 	return a
 end
 
 function matmul(n, a, b)
 	c = zeros(Float64, n, n)
-	for i = 1:n
+	for j = 1:n
 		for k = 1:n
-			aik = a[i,k]
-			for j = 1:n
-				c[i,j] += aik * b[k,j]
+			@inbounds for i = 1:n
+					c[i,j] += a[i,k] * b[k,j]
 			end
 		end
 	end
