Skip to content

Commit 1a3c9c4

Browse files
justinjfuGoogle-ML-Automation
authored andcommitted
[Pallas] Fix type annotation on TritonCompilerParams
PiperOrigin-RevId: 702825912
1 parent db97d7a commit 1a3c9c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

jax/_src/pallas/triton/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,4 +35,4 @@ class TritonCompilerParams(pallas_core.CompilerParams):
3535
PLATFORM: ClassVar[str] = "triton"
3636
num_warps: int | None = None
3737
num_stages: int | None = None
38-
serialized_metadata: str | None = None
38+
serialized_metadata: bytes | None = None

0 commit comments

Comments
 (0)