diff --git a/backends/cuda/cuda_backend.py b/backends/cuda/cuda_backend.py index cc2d662b335..778446bcecb 100644 --- a/backends/cuda/cuda_backend.py +++ b/backends/cuda/cuda_backend.py @@ -49,7 +49,8 @@ def get_decomposition_table(cls) -> Dict[Any, Any]: @classmethod def get_custom_passes(cls) -> List[typing.Any]: """Return CUDA-specific passes: ReplaceEdgeOpWithTritonOpPass""" - return [ReplaceEdgeOpWithTritonOpPass()] + # return [ReplaceEdgeOpWithTritonOpPass()] + return [] @classmethod def get_aoti_compile_options(