cuda

Properties

matmul.allow_tf32

dragon.vm.torch.backends.cuda.matmul.allow_tf32 = False

The flag that allows TF32 math type for matmul or not.

Functions

is_built

cuda.is_built()[source]

Return a bool reporting if built with CUDA support.

Returns:
boolTrue if built otherwise False.