package jax_triton

Get desktop application:
View/edit binary Protocol Buffers messages

• oneof value

  • TritonKernelCall kernel_call = 1

  • TritonAutotunedKernelCall autotuned_kernel_call = 2

• bytes metadata = 3

• string name = 4

  User assigned name.

Used in: TritonAnyKernelCall

• string name = 1

  Name used in auto-tuning log messages.

• repeated TritonAutotunedKernelCall.Config configs = 2

• repeated TritonAutotunedKernelCall.InputOutputAlias input_output_aliases = 3

Used in: TritonAutotunedKernelCall

• optional TritonKernelCall kernel_call = 1

• string description = 2

Used in: TritonAutotunedKernelCall

• uint32 input_buffer_idx = 1

• uint32 output_buffer_idx = 2

• uint64 buffer_size_bytes = 3

Used in: TritonKernelCall

• string kernel_name = 1

  Kernel function name within module.

• uint32 num_warps = 2

• uint32 shared_mem_bytes = 3

• string ptx = 4

• string ttir = 5

• uint32 compute_capability = 6

• uint32 cluster_dim_0 = 7

• uint32 cluster_dim_1 = 8

• uint32 cluster_dim_2 = 9

Used in: TritonAnyKernelCall, TritonAutotunedKernelCall.Config

• optional TritonKernel kernel = 1

• uint32 grid_0 = 2

• uint32 grid_1 = 3

• uint32 grid_2 = 4

• repeated TritonKernelCall.Parameter parameters = 5

Used in: TritonKernelCall

• oneof value

  • Parameter.Array array = 1

  • bool bool_ = 2

  • int32 i32 = 3

  • uint32 u32 = 4

  • int64 i64 = 5

  • uint64 u64 = 6

  • float f32 = 7

  • double f64 = 8

Used in: Parameter

• uint64 bytes_to_zero = 1

• uint64 ptr_divisibility = 2