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.