[fusilli] Add abs pointwise op support (#294)

Adds ABS mode to pointwise operations, mapping to torch.aten.abs.
Includes lit ASM emitter test and sample test coverage.

---------

Signed-off-by: Rob Suderman <rob.suderman@gmail.com>
Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
Co-authored-by: Sambhav Jain <sambhav.eee@gmail.com>

Author: 3 people

include/fusilli/attributes/pointwise_attributes.h

@@ -25,6 +25,7 @@
 namespace fusilli {
 
 #define FUSILLI_POINTWISE_OPS(OP)                                              \
+  OP(ABS)                                                                      \
   OP(ADD)                                                                      \
   /* OP(ADD_SQUARE)  */                                                        \
   /* OP(BINARY_SELECT)  */                                                     \
@@ -127,6 +128,7 @@ inline const std::unordered_map<PointwiseAttr::Mode, std::string>
 
 inline const std::unordered_map<PointwiseAttr::Mode, int>
     PointwiseAttr::kModeToRequiredInputCount = {
+        {PointwiseAttr::Mode::ABS, 1},
         {PointwiseAttr::Mode::ADD, 2},
         {PointwiseAttr::Mode::CEIL, 1},
         {PointwiseAttr::Mode::CMP_EQ, 2},

include/fusilli/support/asm_emitter.h

@@ -1731,6 +1731,7 @@ inline ErrorOr<std::string> PointwiseNode::emitNodePreAsm() const {
   FUSILLI_DECLARE_BINARY_POINTWISE_EMITTER(PWOP, kSubAddSchema, OPIR)
 
   switch (pointwiseAttr.getMode()) {
+    FUSILLI_DECLARE_UNARY_TORCH_EMITTER(ABS, torch.aten.abs)
     FUSILLI_DECLARE_UNARY_TORCH_EMITTER(CEIL, torch.aten.ceil)
     FUSILLI_DECLARE_BINARY_TORCH_EMITTER(CMP_EQ, torch.aten.eq.Tensor)
     FUSILLI_DECLARE_BINARY_TORCH_EMITTER(CMP_LT, torch.aten.lt.Tensor)

samples/pointwise/pointwise_unary_ops.cpp

@@ -40,8 +40,9 @@ TEST_CASE("Pointwise unary ops", "[pointwise][graph]") {
   const auto dim = std::vector<int64_t>{2, 16, 64, 64};
 
   const auto mode =
-      GENERATE(PointwiseAttr::Mode::CEIL, PointwiseAttr::Mode::RELU_FWD,
-               PointwiseAttr::Mode::SIGMOID_FWD, PointwiseAttr::Mode::TANH_FWD);
+      GENERATE(PointwiseAttr::Mode::ABS, PointwiseAttr::Mode::CEIL,
+               PointwiseAttr::Mode::RELU_FWD, PointwiseAttr::Mode::SIGMOID_FWD,
+               PointwiseAttr::Mode::TANH_FWD);
 
   auto execute = [&]<typename T>(Handle &handle, DataType dt, T x) {
     auto buildNewGraph = [&](Handle &handleArg) {
@@ -96,6 +97,11 @@ TEST_CASE("Pointwise unary ops", "[pointwise][graph]") {
     // Calculate reference value
     T y = 0;
     switch (mode) {
+    case PointwiseAttr::Mode::ABS: {
+      double xD = static_cast<double>(x);
+      y = std::abs(xD);
+      break;
+    }
     case PointwiseAttr::Mode::RELU_FWD: {
       y = std::max(x, T(0));
       break;

tests/CMakeLists.txt

@@ -141,6 +141,7 @@ add_fusilli_lit_tests(
     lit/test_conv_asm_emitter_ndhwc_kdrsc.cpp
     lit/test_conv_asm_emitter_ndhwc_kdrsc_grouped.cpp
     lit/test_conv_asm_emitter_nhwc_krsc_grouped.cpp
+    lit/test_pointwise_asm_emitter_abs.cpp
     lit/test_pointwise_asm_emitter_relu.cpp
     lit/test_pointwise_asm_emitter_add.cpp
     lit/test_pointwise_asm_emitter_add_transposed.cpp

tests/lit/test_pointwise_asm_emitter_abs.cpp

@@ -0,0 +1,60 @@
+// Copyright 2026 Advanced Micro Devices, Inc.
+//
+// Licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+// RUN: %{TEST_EXE} | iree-opt --verify-roundtrip
+// RUN: %{TEST_EXE} | FileCheck %s --check-prefix=TORCH-CHECK
+// RUN: %{TEST_EXE} stats | FileCheck %s --check-prefix=%{BACKEND}-STATS-CHECK
+
+// clang-format off
+//
+// TORCH-CHECK:   module @module {
+// TORCH-CHECK:     func.func @main(%result_: !torch.tensor<[16,256,64,32],f32>, %arg0: !torch.vtensor<[16,256,64,32],f32>) attributes {torch.assume_strict_symbolic_shapes} {
+// TORCH-CHECK:       %permute_IN_0_val_0_pointwise_abs = torch.constant.int 0
+// TORCH-CHECK:       %permute_IN_0_val_1_pointwise_abs = torch.constant.int 1
+// TORCH-CHECK:       %permute_IN_0_val_2_pointwise_abs = torch.constant.int 2
+// TORCH-CHECK:       %permute_IN_0_val_3_pointwise_abs = torch.constant.int 3
+// TORCH-CHECK:       %permute_IN_0_pointwise_abs = torch.prim.ListConstruct %permute_IN_0_val_0_pointwise_abs, %permute_IN_0_val_1_pointwise_abs, %permute_IN_0_val_2_pointwise_abs, %permute_IN_0_val_3_pointwise_abs : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int>
+// TORCH-CHECK:       %arg0_pointwise_abs_perm = torch.aten.permute %arg0, %permute_IN_0_pointwise_abs : !torch.vtensor<[16,256,64,32],f32>, !torch.list<int> -> !torch.vtensor<[16,256,64,32],f32>
+// TORCH-CHECK:       %result_pointwise_abs_perm = torch.aten.abs %arg0_pointwise_abs_perm : !torch.vtensor<[16,256,64,32],f32> -> !torch.vtensor<[16,256,64,32],f32>
+// TORCH-CHECK:       %permute_OUT_0_val_0_pointwise_abs = torch.constant.int 0
+// TORCH-CHECK:       %permute_OUT_0_val_1_pointwise_abs = torch.constant.int 1
+// TORCH-CHECK:       %permute_OUT_0_val_2_pointwise_abs = torch.constant.int 2
+// TORCH-CHECK:       %permute_OUT_0_val_3_pointwise_abs = torch.constant.int 3
+// TORCH-CHECK:       %permute_OUT_0_pointwise_abs = torch.prim.ListConstruct %permute_OUT_0_val_0_pointwise_abs, %permute_OUT_0_val_1_pointwise_abs, %permute_OUT_0_val_2_pointwise_abs, %permute_OUT_0_val_3_pointwise_abs : (!torch.int, !torch.int, !torch.int, !torch.int) -> !torch.list<int>
+// TORCH-CHECK:       %result = torch.aten.permute %result_pointwise_abs_perm, %permute_OUT_0_pointwise_abs : !torch.vtensor<[16,256,64,32],f32>, !torch.list<int> -> !torch.vtensor<[16,256,64,32],f32>
+// TORCH-CHECK:       torch.overwrite.tensor.contents %result overwrites %result_ : !torch.vtensor<[16,256,64,32],f32>, !torch.tensor<[16,256,64,32],f32>
+// TORCH-CHECK:       return
+// TORCH-CHECK:     }
+// TORCH-CHECK:   }
+//
+// AMDGPU-STATS-CHECK: "transient-memory-size": 0
+// AMDGPU-STATS-CHECK: "dispatch-count": 1
+// CPU-STATS-CHECK: "transient-memory-size": 0
+// CPU-STATS-CHECK: "dispatch-count": 1
+//
+// clang-format on
+
+#include <fusilli.h>
+
+#include "utils.h"
+
+#include <iostream>
+#include <string>
+
+using namespace fusilli;
+
+int main(int argc, char **argv) {
+  std::string mode = (argc > 1) ? argv[1] : "default";
+
+  auto status = testUnaryPointwiseAsmEmitter(
+      "pointwise_asm_emitter_abs", "pointwise_abs", mode,
+      PointwiseAttr::Mode::ABS, {16, 256, 64, 32});
+  if (isError(status)) {
+    std::cerr << "Test failed: " << status << std::endl;
+    return 1;
+  }
+  return 0;
+}