Note: uncommented `test_gemm`, since backend has an issue with swizzling. This will get uncommented in a subsequent PR.