Are the tensor contractions in "High-Performance Tensor Contractions for GPUs" integrated into MAGMA?

[Yakup Budanaz]

Hi,

I wanted to quickly ask: are the tensor contractions introduced in the work High-Performance Tensor Contractions for GPUs integrated into MAGMA? I checked the documentation, but could not find the routines on tensor contractions in the documentation (2.7.1). The future-work mentioned that the implementation would be incorporated into MAGMA, and I wanted to double-check.

Best Regards,

Yakup Budanaz

[Ahmad Abdelfattah]

Hi Yakup,

The results in Figure 3 of this paper are obtained by running the kernel under magmablas/zgemm_batched_smallsq.cu. This kernel performs matrix multiplications on very small square matrices, which can be used for tensor contractions. The code is written for the double complex precision, which is used to auto-generate other precisions when magma is built. 

Ahmad

--
You received this message because you are subscribed to the Google Groups "MAGMA User" group.
To unsubscribe from this group and stop receiving emails from it, send an email to magma-user+...@icl.utk.edu.
To view this discussion on the web visit https://groups.google.com/a/icl.utk.edu/d/msgid/magma-user/ae6064d9-c878-4604-a33c-292a0fcbe72dn%40icl.utk.edu.