diff --git a/torchbenchmark/util/extra_args.py b/torchbenchmark/util/extra_args.py
index 7e84986426..84975b3923 100644
--- a/torchbenchmark/util/extra_args.py
+++ b/torchbenchmark/util/extra_args.py
@@ -1,5 +1,6 @@
 import argparse
 import enum
+import warnings
 from typing import List, Optional, Tuple
 
 from torchbenchmark.util.backends import BACKENDS, list_backends