Disableing Tensor core

[aran nokan]

Hi,

I am trying to run my code with CUDA core, so I used this line to change the mode, but by profiling (nsys) the execution I am seeing that the cutlass tensor core kernel is running in both queues.

cublasSetMathMode(queues[0]->cublas_handle(), CUBLAS_DEFAULT_MATH);

cublasSetMathMode(queues[1]->cublas_handle(), CUBLAS_DEFAULT_MATH);

Do we have something inside of the GEMM that enables Tensor core again?

Best regards,

A. N.

[Ahmad Abdelfattah]

If you are referring to the magma_hgemm routine, then yes, the routine does enable the tensor cores by default. 

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/CAKHt_YYV1v7hj9x%3DJsCCe-ewjEU3KsNqmTJ14JnXd03mV8N_tg%40mail.gmail.com.