This is the source code used for NDS: N-Dimensional Storage
make
-
cuda_samples: Copied from cuda samples that helps to learn TCU wmma -
data: Contains matrix generator in sequential/NDS format -
lib:gemm: offers APIs binding cublasGemm with NDS formatspdkrpc: communicates to SPDK app to fetch matrix from remote servertensorstore: converts data format between sequential and NDS formats
-
test:
cublas_mm_verification: Verify the resulting matrix is correct by using cublasGEMMmatrix_transpose: Verify and check the performance of fetching data from SPDK transposing matrix.mm_verification: Verify the resulting matrix is correct by using blockmm on CPUread_test: Verify tensorstore functionalitiestensorstore_mm_perftest: Performance test of APIs from gemm librarytensorstore_mm_verification: Verifications of APIs from gemm librarytranspose_perftest: Performance test for cublasGEMM after applying normal/transposed matrix operands