add setup to pyproject.toml

Signed-off-by: Phuong Nguyen <[email protected]>
NVIDIA · Jul 2, 2024 · 8f5e77a · 8f5e77a
1 parent 8df0e08
commit 8f5e77a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -6,5 +6,9 @@
 requires = [ 
   "cmake>=3.18",
   "ninja", 
-  "pybind11"
+  "pybind11",
+  "setuptools",
+  "transformer-engine"
   ]
+
+transformer-engine = "setup"