Skip to main content

Mojo module

io_spec

Aliases

FusedInput

alias FusedInput = IOSpec[False, IO(2)]()

FusedOutput

alias FusedOutput = IOSpec[True, IO(3)]()

Input

alias Input = IOSpec[False, IO(1)]()

IOUnknown

alias IOUnknown = IOSpec[True, IO(-1)]()

MutableInput

alias MutableInput = IOSpec[True, IO(1)]()

Output

alias Output = IOSpec[True, IO(0)]()

Structs

  • IO:
  • IOSpec: Parameter used to encode whether a particular tensor argument to a DPS kernel is an output, input, or mutable input.

Was this page helpful?