[all-commits] [llvm/llvm-project] f35318: [mlir][amdgpu] Add support for multi-dim arith.tru...

[Rob Suderman via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f35318e828d166bee04a117c1a87c1c0e32a57da
      https://github.com/llvm/llvm-project/commit/f35318e828d166bee04a117c1a87c1c0e32a57da
  Author: Rob Suderman <rob.suderman at gmail.com>
  Date:   2024-07-09 (Tue, 09 Jul 2024)

  Changed paths:
    M mlir/lib/Conversion/ArithToAMDGPU/ArithToAMDGPU.cpp
    M mlir/test/Conversion/ArithToAMDGPU/8-bit-floats.mlir

  Log Message:
  -----------
  [mlir][amdgpu] Add support for multi-dim arith.truncf/extf fp8 lowering (#98074)

The existing `fp8` lowering from `arith` to `amdgpu` bails out on the
multidimensional case. We can handle this by `vector.shape_cast`
collapsing to the 1-D case on extraction and re-casting back to the
desired output shape.



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