Skip to content

Commit fded6bd

Browse files
loisloGoogle-ML-Automation
authored andcommitted
Follow up pr after triton integration cl and tokamax pr.
Remove obsolete cluster dimensions from TritonKernel proto. PiperOrigin-RevId: 843226944
1 parent ae9a27e commit fded6bd

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

jaxlib/gpu/triton.proto

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,8 @@ message TritonKernel {
1010
string ptx = 4;
1111
string ttir = 5;
1212
uint32 compute_capability = 6;
13-
uint32 cluster_dim_0 = 7;
14-
uint32 cluster_dim_1 = 8;
15-
uint32 cluster_dim_2 = 9;
13+
14+
reserved 7, 8, 9;
1615
}
1716

1817
message TritonKernelCall {

0 commit comments

Comments
 (0)