[all-commits] [llvm/llvm-project] 8a57d8: [mlir] Add Scalar Broadcast TOSA Depthwise Conv (#...

[Jack Frankland via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 8a57d8212000a6611b24f79aa4d28af17cef7150
      https://github.com/llvm/llvm-project/commit/8a57d8212000a6611b24f79aa4d28af17cef7150
  Author: Jack Frankland <jack.frankland at arm.com>
  Date:   2024-10-03 (Thu, 03 Oct 2024)

  Changed paths:
    M mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp
    M mlir/test/Conversion/TosaToLinalg/tosa-to-linalg-named.mlir

  Log Message:
  -----------
  [mlir] Add Scalar Broadcast TOSA Depthwise Conv (#110806)

Support broadcasting of depthwise conv2d bias in tosa->linalg named
lowering in the case that bias is a rank-1 tensor with exactly 1
element. In this case TOSA specifies the value should first be broadcast
across the bias dimension and then across the result tensor.

Add `lit` tests for depthwise conv2d with scalar bias and for conv3d
which was already supported but missing coverage.

Signed-off-by: Jack Frankland <jack.frankland at arm.com>



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications