Skip to content

Commit c5f5ac1

Browse files
Revert "[Backend] Improve dot support to target FMA (#4516)"
This reverts commit 52cf1ae.
1 parent 536b0b9 commit c5f5ac1

File tree

13 files changed

+294
-573
lines changed

13 files changed

+294
-573
lines changed

include/triton/Conversion/TritonGPUToLLVM/Utility.h

Lines changed: 0 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -335,18 +335,12 @@ SmallVector<Value> delinearize(RewriterBase &rewriter, Location loc,
335335
SmallVector<Value> delinearize(RewriterBase &rewriter, Location loc,
336336
Value linear, ArrayRef<unsigned> shape);
337337

338-
SmallVector<unsigned> delinearize(unsigned linear, ArrayRef<unsigned> shape,
339-
ArrayRef<unsigned> order);
340-
341338
Value linearize(RewriterBase &rewriter, Location loc, ArrayRef<Value> multiDim,
342339
ArrayRef<unsigned> shape, ArrayRef<unsigned> order);
343340

344341
Value linearize(RewriterBase &rewriter, Location loc, ArrayRef<Value> multiDim,
345342
ArrayRef<unsigned> shape);
346343

347-
size_t linearize(ArrayRef<unsigned> multiDim, ArrayRef<unsigned> shape,
348-
ArrayRef<unsigned> order);
349-
350344
Value addStringToModule(Location loc, RewriterBase &rewriter, StringRef key,
351345
StringRef content);
352346

@@ -501,24 +495,6 @@ inline Value dot(RewriterBase &rewriter, Location loc, ArrayRef<Value> offsets,
501495
return ret;
502496
}
503497

504-
/// Extend 2d shared object to 3d.
505-
///
506-
/// If tensor has 3 dimensions, returns original shared object.
507-
/// If tensor shape is [M, N], return shared object describing shape [1, M, N]
508-
///
509-
/// This Function is used to simplify processing of 2d and 3d dot operands,
510-
/// particularly in the conversion of local_load operation.
511-
///
512-
/// \param rewriter
513-
/// \param loc
514-
/// \param smemObj
515-
/// \param shape shape of a tensor represented by smemObj
516-
/// \returns shared object describing 3d tensor
517-
SharedMemoryObject
518-
getExpandedSharedMemoryObject(ConversionPatternRewriter &rewriter, Location loc,
519-
SharedMemoryObject smemObj,
520-
ArrayRef<int64_t> shape);
521-
522498
// -----------------------------------------------------------------------
523499
// Blocked layout indices
524500
// -----------------------------------------------------------------------

include/triton/Dialect/TritonGPU/IR/Dialect.h

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -234,12 +234,6 @@ void dumpHWLayout(RankedTensorType tensorType);
234234
// Return a string representation of the layout of the tensor.
235235
std::string getLayoutStr(RankedTensorType tensorType, bool useHWPointOfView);
236236

237-
template <typename T>
238-
llvm::SmallVector<T> expandMatrixShapeWithBatch(llvm::ArrayRef<T> s);
239-
240-
llvm::SmallVector<unsigned>
241-
expandMatrixOrderWithBatch(llvm::ArrayRef<unsigned> o);
242-
243237
} // namespace gpu
244238
} // namespace triton
245239
} // namespace mlir

0 commit comments

Comments
 (0)