taichi.aot.conventions.gfxruntime140.sr#

Structured representation of all JSON data structures following the GfxRuntime140.

class taichi.aot.conventions.gfxruntime140.sr.Argument(name: str | None)#

Bases: abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

class taichi.aot.conventions.gfxruntime140.sr.ArgumentNdArray(name: str | None, dtype: DataType, element_shape: List[int], ndim: int, access: NdArrayAccess)#

Bases: Argument

Helper class that provides a standard way to create an ABC using inheritance.

class taichi.aot.conventions.gfxruntime140.sr.ArgumentRwTexture(name: str | None, fmt: taichi.Format, ndim: int)#

Bases: Argument

Helper class that provides a standard way to create an ABC using inheritance.

class taichi.aot.conventions.gfxruntime140.sr.ArgumentScalar(name: str | None, dtype: DataType)#

Bases: Argument

Helper class that provides a standard way to create an ABC using inheritance.

class taichi.aot.conventions.gfxruntime140.sr.ArgumentTexture(name: str | None, ndim: int)#

Bases: Argument

Helper class that provides a standard way to create an ABC using inheritance.

class taichi.aot.conventions.gfxruntime140.sr.BufferBinding(binding: int, iarg: int, buffer_bind_ty: BufferBindingType)#

class taichi.aot.conventions.gfxruntime140.sr.BufferBindingType#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

Args = 2#

ExtArr = 5#

GlobalTmps = 1#

ListGen = 4#

Rets = 3#

Root = 0#

name(self)#

The name of the Enum member.

value(self)#

The value of the Enum member.

class taichi.aot.conventions.gfxruntime140.sr.Context(args: List[Argument], ret: ReturnValue | None)#

class taichi.aot.conventions.gfxruntime140.sr.DataType#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

f16 = 0#

f32 = 1#

f64 = 2#

i16 = 4#

i32 = 5#

i64 = 6#

i8 = 3#

u16 = 9#

u32 = 10#

u64 = 11#

u8 = 8#

name(self)#

The name of the Enum member.

value(self)#

The value of the Enum member.

class taichi.aot.conventions.gfxruntime140.sr.Dispatch(kernel: Kernel, args: List[NamedArgument])#

class taichi.aot.conventions.gfxruntime140.sr.Field(name: str, dtype: DataType, element_shape: List[int], shape: List[int], offset: int)#

class taichi.aot.conventions.gfxruntime140.sr.Graph(name: str, dispatches: List[Dispatch])#

class taichi.aot.conventions.gfxruntime140.sr.Kernel(name: str, context: Context, tasks: List[Task])#

class taichi.aot.conventions.gfxruntime140.sr.LaunchGrid(block_size: int, grid_size: int)#

class taichi.aot.conventions.gfxruntime140.sr.Metadata(fields: List[Field], kernels: List[Kernel], required_caps: List[taichi.DeviceCapability], root_buffer_size: int)#

class taichi.aot.conventions.gfxruntime140.sr.NamedArgument(name: str, arg: Argument)#

class taichi.aot.conventions.gfxruntime140.sr.NdArrayAccess#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

NoAccess = 0#

Read = 1#

ReadWrite = 3#

Write = 2#

name(self)#

The name of the Enum member.

value(self)#

The value of the Enum member.

class taichi.aot.conventions.gfxruntime140.sr.ReturnValue(dtype: DataType)#

class taichi.aot.conventions.gfxruntime140.sr.Task(name: str, task_ty: TaskType, buffer_binds: List[BufferBinding], texture_binds: List[TextureBinding], launch_grid: LaunchGrid)#

class taichi.aot.conventions.gfxruntime140.sr.TaskType#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

Gc = 5#

GcRc = 6#

ListGen = 4#

MeshFor = 3#

RangeFor = 1#

Serial = 0#

StructFor = 2#

name(self)#

The name of the Enum member.

value(self)#

The value of the Enum member.

class taichi.aot.conventions.gfxruntime140.sr.TextureBinding(binding: int, iarg: int, texture_bind_ty: TextureBindingType)#

class taichi.aot.conventions.gfxruntime140.sr.TextureBindingType#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

RwTexture = 1#

Texture = 0#

name(self)#

The name of the Enum member.

value(self)#

The value of the Enum member.

taichi.aot.conventions.gfxruntime140.sr.from_dr_field(d: taichi.aot.conventions.gfxruntime140.dr.FieldAttributes) → Field#

taichi.aot.conventions.gfxruntime140.sr.from_dr_graph(meta: Metadata, j: taichi.aot.conventions.gfxruntime140.dr.Graph) → Graph#

taichi.aot.conventions.gfxruntime140.sr.from_dr_kernel(d: taichi.aot.conventions.gfxruntime140.dr.KernelAttributes) → Kernel#

taichi.aot.conventions.gfxruntime140.sr.from_dr_metadata(d: taichi.aot.conventions.gfxruntime140.dr.Metadata) → Metadata#

taichi.aot.conventions.gfxruntime140.sr.get_data_type_size(dtype: DataType) → int#

taichi.aot.conventions.gfxruntime140.sr.to_dr_field(f: Field) → Dict[str, Any]#

taichi.aot.conventions.gfxruntime140.sr.to_dr_graph(s: Graph) → taichi.aot.conventions.gfxruntime140.dr.Graph#

taichi.aot.conventions.gfxruntime140.sr.to_dr_kernel(s: Kernel) → Dict[str, Any]#

taichi.aot.conventions.gfxruntime140.sr.to_dr_metadata(s: Metadata) → taichi.aot.conventions.gfxruntime140.dr.Metadata#