From 47c1ddeca03ca3cc416f6b73370ecb2d189301d6 Mon Sep 17 00:00:00 2001
From: pytorchbot <soumith+bot@pytorch.org>
Date: Wed, 18 Dec 2024 15:57:21 -0600
Subject: [PATCH] Fix wheel build CI jobs

ghstack-source-id: f077c82325da4211cd16705e812e6ebed6713263
Pull Request resolved: https://github.com/pytorch/executorch/pull/7367

Co-authored-by: Stephen Jia <ssjia@meta.com>
(cherry picked from commit 18142f758ac278e37d4b90849d7d47661de0509b)
---
 build/packaging/smoke_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/build/packaging/smoke_test.py b/build/packaging/smoke_test.py
index e6ee2380ae..1573e37bf5 100644
--- a/build/packaging/smoke_test.py
+++ b/build/packaging/smoke_test.py
@@ -17,7 +17,7 @@
 
 # Import custom ops. This requires portable_lib to be loaded first.
 from executorch.extension.llm.custom_ops import (  # noqa: F401, F403
-    sdpa_with_kv_cache,
+    custom_ops,
 )  # usort: skip
 
 # Import quantized ops. This requires portable_lib to be loaded first.