[Inductor] Fix cuda_kernel typing

mlazos · mlazos · commit ccf0adb2ab17 · 2025-04-28T17:18:04.000-07:00
ghstack-source-id: e61ab35 Pull Request resolved: #150908
diff --git a/torch/_inductor/codegen/cuda/cuda_kernel.py b/torch/_inductor/codegen/cuda/cuda_kernel.py
@@ -1,4 +1,5 @@
 # mypy: allow-untyped-defs
+import functools
 import logging
 from dataclasses import dataclass
 from typing import Any, Callable, Literal, Optional, TYPE_CHECKING, Union

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`# mypy: allow-untyped-defs`
	`2`	`+import functools`
`2`	`3`	`import logging`
`3`	`4`	`from dataclasses import dataclass`
`4`	`5`	`from typing import Any, Callable, Literal, Optional, TYPE_CHECKING, Union`