Custom Triton kernel registration · pytorch/executorch@175bfa6

Do not call assert False since python -O removes these calls. Instead callers should raise AssertionError(). See https://github.com/PyCQA/flake8-bugbear#list-of-warnings.

FLAKE8 F401: /pytorch/executorch/custom_triton_playground/test-sdpa-with-custom-kernel.py#L15

'torch.export.Dim' imported but unused See https://www.flake8rules.com/rules/F401.html.

FLAKE8 E741: /pytorch/executorch/custom_triton_playground/sdpa_triton.py#L242

ambiguous variable name 'O' See https://www.flake8rules.com/rules/E741.html.

FLAKE8 F401: /pytorch/executorch/custom_triton_playground/sdpa_triton.py#L3

'typing.Any' imported but unused See https://www.flake8rules.com/rules/F401.html.

UFMT format: /pytorch/executorch/backends/cuda/cuda_backend.py#L0

Run `lintrunner -a` to apply this patch.

FLAKE8 F401: /pytorch/executorch/backends/cuda/cuda_backend.py#L30

'torch.nn.attention.SDPBackend' imported but unused See https://www.flake8rules.com/rules/F401.html.

FLAKE8 F401: /pytorch/executorch/custom_triton_playground/optimized_sdpa_triton.py#L2

'typing.Any' imported but unused See https://www.flake8rules.com/rules/F401.html.

Provide feedback