Skip to main content

/

Docs

Nightly

v25.1

Log in

Mojo module

info

Contains information about the GPUs.

Aliases​

  • ​A10 = Info("A10", Vendor(__init__[__mlir_type.!kgen.int_literal](2)), "cuda", "ampere", "nvptx-short-ptr=true", __init__[__mlir_type.!kgen.float_literal](8.600000e+00), "sm_86", 72, 32, 1536, 32, 64, 2048, 32, 102400, 65536, 256, "warp", 255, 65536, 16, 128, 4, 1024, Flops(__init__[__mlir_type.!kgen.int_literal](125), __init__[__mlir_type.!kgen.int_literal](250), __init__[__mlir_type.!kgen.int_literal](500), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.float_literal](6.250000e+01), __init__[__mlir_type.!kgen.int_literal](0))):
  • ​A100 = Info("A100", Vendor(__init__[__mlir_type.!kgen.int_literal](2)), "cuda", "ampere", "nvptx-short-ptr=true", __init__[__mlir_type.!kgen.float_literal](8.000000e+00), "sm_80", 108, 32, 2048, 32, 64, 2048, 32, 167936, 65536, 256, "warp", 255, 65536, 32, 128, 4, 1024, Flops(__init__[__mlir_type.!kgen.int_literal](312), __init__[__mlir_type.!kgen.int_literal](624), __init__[__mlir_type.!kgen.int_literal](1248), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.int_literal](156), __init__[__mlir_type.!kgen.float_literal](1.950000e+01))):
  • ​DEFAULT_GPU = from_name[::StringLiteral]():
  • ​DEFAULT_GPU_ARCH = _accelerator_arch():
  • ​DEFAULT_GPU_TARGET = from_name[::StringLiteral]().target[::Int]():
  • ​H100 = Info("H100", Vendor(__init__[__mlir_type.!kgen.int_literal](2)), "cuda", "hopper", "nvptx-short-ptr=true", __init__[__mlir_type.!kgen.float_literal](9.000000e+00), "sm_90a", 132, 32, -1, 32, 64, 2048, 32, 233472, 65536, 256, "warp", 255, 65536, 32, 128, 4, 1024, Flops(__init__[__mlir_type.!kgen.int_literal](1979), __init__[__mlir_type.!kgen.int_literal](3958), __init__[__mlir_type.!kgen.int_literal](7916), __init__[__mlir_type.!kgen.int_literal](3958), __init__[__mlir_type.!kgen.int_literal](989), __init__[__mlir_type.!kgen.int_literal](67))):
  • ​L4 = Info("L4", Vendor(__init__[__mlir_type.!kgen.int_literal](2)), "cuda", "ada", "nvptx-short-ptr=true", __init__[__mlir_type.!kgen.float_literal](8.900000e+00), "sm_89", 58, 32, 1536, 32, 64, 2048, 32, 102400, 65536, 256, "warp", 255, 65536, 24, 128, 4, 1024, Flops(__init__[__mlir_type.!kgen.int_literal](121), __init__[__mlir_type.!kgen.int_literal](242), __init__[__mlir_type.!kgen.int_literal](485), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.int_literal](0))):
  • ​MI300X = Info("MI300X", Vendor(__init__[__mlir_type.!kgen.int_literal](1)), "hip", "gfx942", "", __init__[__mlir_type.!kgen.float_literal](9.400000e+00), "CDNA3", 304, 64, 2048, 64, 32, 2048, 2, 65536, 65536, 256, "warp", 255, 65536, 2, 128, 4, 1024, Flops(__init__[__mlir_type.!kgen.float_literal](1.307400e+03), __init__[__mlir_type.!kgen.float_literal](2.614900e+03), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.float_literal](1.307400e+03), __init__[__mlir_type.!kgen.float_literal](6.537000e+02), __init__[__mlir_type.!kgen.int_literal](0))):
  • ​NoGPU = Info("NoGPU", Vendor(__init__[__mlir_type.!kgen.int_literal](0)), "none", "no_gpu", "", __init__[__mlir_type.!kgen.int_literal](0), "", 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, "none", 0, 0, 0, 0, 0, 0, Flops(__init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.int_literal](0), __init__[__mlir_type.!kgen.int_literal](0))):

Structs​

Functions​