diff --git a/fbgemm_gpu/experimental/gemm/triton/fp8_gemm_benchmark.py b/fbgemm_gpu/experimental/gemm/triton/fp8_gemm_benchmark.py index 1400c09fa..1f178a46b 100644 --- a/fbgemm_gpu/experimental/gemm/triton/fp8_gemm_benchmark.py +++ b/fbgemm_gpu/experimental/gemm/triton/fp8_gemm_benchmark.py @@ -4,6 +4,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-strict + from typing import Callable, Tuple import torch diff --git a/fbgemm_gpu/experimental/gemm/triton/fp8_gemm_test.py b/fbgemm_gpu/experimental/gemm/triton/fp8_gemm_test.py index 2d94116ca..ec2624a94 100644 --- a/fbgemm_gpu/experimental/gemm/triton/fp8_gemm_test.py +++ b/fbgemm_gpu/experimental/gemm/triton/fp8_gemm_test.py @@ -4,6 +4,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-strict + import unittest from typing import Optional, Tuple