wip

joelberkeley · Jan 1, 2025 · 3b7dcc1 · 3b7dcc1
1 parent 3660166
commit 3b7dcc1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/XLA_VERSION b/XLA_VERSION
@@ -1 +1 @@
-2fb20601f1cc6cab7f29f8bc73d90cd31e74bba0
+46bd34518db14456697224b9684f47bf4cdcf6ba
diff --git a/spidr/backend/src/xla/hlo/builder/xla_builder.cpp b/spidr/backend/src/xla/hlo/builder/xla_builder.cpp
@@ -338,12 +338,7 @@ extern "C" {
     }
 
     XlaOp* Abs(XlaOp& operand) { return unaryOp(xla::Abs, operand); }
-
-    XlaOp* Exp(XlaOp& operand) {
-        xla::XlaOp res = xla::Exp(reinterpret_cast<xla::XlaOp&>(operand));
-        return reinterpret_cast<XlaOp*>(new xla::XlaOp(res));
-    }
-
+    XlaOp* Exp(XlaOp& operand) { return unaryOp(xla::Exp, operand); }
     XlaOp* Floor(XlaOp& operand) { return unaryOp(xla::Floor, operand); }
     XlaOp* Ceil(XlaOp& operand) { return unaryOp(xla::Ceil, operand); }
     XlaOp* Log(XlaOp& operand) { return unaryOp(xla::Log, operand); }
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		2fb20601f1cc6cab7f29f8bc73d90cd31e74bba0
		46bd34518db14456697224b9684f47bf4cdcf6ba