set_cublas_flags¶

dragon.cuda.set_cublas_flags(allow_tf32=None)[source]¶

Set the flags of cuBLAS library.

Parameters:

• allow_tf32 (bool, optional, default=False) – Allow TF32 tensor core operation or not.