Skip to content

Commit af7a31f

Browse files
superbobryGoogle-ML-Automation
authored andcommitted
[pallas:triton] Fixed a typo in a type annotation
PiperOrigin-RevId: 707905188
1 parent 7680532 commit af7a31f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

jax/_src/pallas/triton/lowering.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ def _check_tensor_size(shape: tuple[int | pallas_core.Mapped, ...]):
273273
def lower_jaxpr_to_triton_module(
274274
jaxpr: jax_core.Jaxpr,
275275
grid_mapping: GridMapping,
276-
name_and_src_info: pallas_core.NameAndStrInfo,
276+
name_and_src_info: pallas_core.NameAndSrcInfo,
277277
platform: str
278278
) -> LoweringResult:
279279
if grid_mapping.num_dynamic_grid_bounds:

0 commit comments

Comments
 (0)