Update thunder/dynamo/compiler_graph_benchmark.py

Co-authored-by: Masaki Kozuki <mkozuki@nvidia.com>
Lightning-AI · Oct 10, 2024 · ec3ab08 · ec3ab08
1 parent ed659d5
commit ec3ab08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thunder/dynamo/compiler_graph_benchmark.py b/thunder/dynamo/compiler_graph_benchmark.py
@@ -45,7 +45,7 @@ def __init__(
                 # script.py
                 import torch
                 import thunder
-                from thunder.benchmarks import ThunderCompilerGraphBenchmarking
+                from thunder.dynamo import ThunderCompilerGraphBenchmarking
 
                 def func(x):
                     x = torch.sin(x)