
Perform 3D tensor-matrix multiplication of two tensors:
$$
C[i][j][l] = \sum_{k=0}^{K-1} A[i][j][k] \cdot B[k][l]
$$

## Input
- Tensor $A$ of size $N \times M \times K$
- Matrix $B$ of size $K \times L$

## Output
- Tensor $C$ of size $N \times M \times L$

## Notes:
- All tensors $\text{A}$, $\text{B}$, and $\text{C}$ are stored in row-major order
- This problem is adapted from [KernelBench](https://github.com/ScalingIntelligence/KernelBench/blob/main/KernelBench/level1/10_3D_tensor_matrix_multiplication.py)

Rank	User	GFLOPS	Runtime (ms)	GPU	Language	Date
#1	ell-hol	109.33T	90.35	H100	CUDA C++	about 2 months ago
#2	sagarreddypatil	50.73T	197.44	B200	Triton	26 days ago
#3	sagarreddypatil	41.44T	244.25	H100	Triton	2 months ago
#4	harmya	6.43T	1538.43	H100	CUDA C++	3 months ago
#5	Firojpaudel	6.08T	1994.18	H100	CUDA C++	3 months ago
#6	ShlokVFX	5.34T	1853.45	H100	CUDA C++	3 months ago