Skip to main content

/

Docs

Nightly

v24.6

Log in

Mojo module

info

Contains information about the GPUs.

Aliases​

  • ​A10 = Info("A10", Vendor(SIMD(2)), "cuda", "ampere", "nvptx-short-ptr=true", SIMD(#kgen.float_literal<43|5>), "sm_86", 72, 32, 1536, 32, 64, 2048, 32, 102400, 65536, 256, "warp", 255, 65536, 16, 128, 4, 1024, Flops(SIMD(125), SIMD(250), SIMD(500), SIMD(0), SIMD(#kgen.float_literal<125|2>), SIMD(0))):
  • ​A100 = Info("A100", Vendor(SIMD(2)), "cuda", "ampere", "nvptx-short-ptr=true", SIMD(#kgen.float_literal<8|1>), "sm_80", 108, 32, 2048, 32, 64, 2048, 32, 167936, 65536, 256, "warp", 255, 65536, 32, 128, 4, 1024, Flops(SIMD(312), SIMD(624), SIMD(1248), SIMD(0), SIMD(156), SIMD(#kgen.float_literal<39|2>))):
  • ​DEFAULT_GPU = from_name[::StringLiteral]():
  • ​DEFAULT_GPU_ARCH = accelerator_arch():
  • ​DEFAULT_GPU_TARGET = from_name[::StringLiteral]().target[::Int]():
  • ​H100 = Info("H100", Vendor(SIMD(2)), "cuda", "hopper", "nvptx-short-ptr=true", SIMD(#kgen.float_literal<9|1>), "sm_90a", 132, 32, -1, 32, 64, 2048, 32, 233472, 65536, 256, "warp", 255, 65536, 32, 128, 4, 1024, Flops(SIMD(1979), SIMD(3958), SIMD(7916), SIMD(3958), SIMD(989), SIMD(67))):
  • ​L4 = Info("L4", Vendor(SIMD(2)), "cuda", "ada", "nvptx-short-ptr=true", SIMD(#kgen.float_literal<89|10>), "sm_89", 58, 32, 1536, 32, 64, 2048, 32, 102400, 65536, 256, "warp", 255, 65536, 24, 128, 4, 1024, Flops(SIMD(121), SIMD(242), SIMD(485), SIMD(0), SIMD(0), SIMD(0))):
  • ​MI300X = Info("MI300X", Vendor(SIMD(1)), "hip", "gfx942", "", SIMD(#kgen.float_literal<47|5>), "CDNA3", 304, 64, 2048, 64, 32, 2048, 2, 65536, 65536, 256, "warp", 255, 65536, 2, 128, 4, 1024, Flops(SIMD(#kgen.float_literal<6537|5>), SIMD(#kgen.float_literal<26149|10>), SIMD(0), SIMD(#kgen.float_literal<6537|5>), SIMD(#kgen.float_literal<6537|10>), SIMD(0))):
  • ​NoGPU = Info("NoGPU", Vendor(SIMD(0)), "none", "no_gpu", "", SIMD(0), "", 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, "none", 0, 0, 0, 0, 0, 0, Flops(SIMD(0), SIMD(0), SIMD(0), SIMD(0), SIMD(0), SIMD(0))):

Structs​

Functions​