Add the missed torch import.

microsoft · Mar 6, 2024 · bed775e · bed775e
1 parent f8454a0
commit bed775e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/op_builder/inference_core_ops.py b/op_builder/inference_core_ops.py
@@ -57,6 +57,8 @@ def get_prefix(self):
  return "deepspeed" if os.path.isdir(ds_path) else ".."
 
  def sources(self):
+ import torch
+
  sources = [
  "inference/v2/kernels/core_ops/core_ops.cpp",
  "inference/v2/kernels/core_ops/bias_activations/bias_activation.cpp",