llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch/Transforms
History
zjgarvey 8995c90879
[TorchToLinalg] add support for quantized group conv (#3341) 
This addresses 7 of the model failures I'm seeing in the test suite. See
[Shark-Turbine issue
#566](https://github.com/nod-ai/SHARK-Turbine/issues/566).

Need the op ```linalg.conv_2d_ngchw_gfchw_q``` to be added upstream
before merging this. See [llvm-project PR #92136
](https://github.com/llvm/llvm-project/pull/92136).

A small additional expansion to operand quantization is included in this
patch to address a model failure that occurs when unblocking the
quantized group convolutions in one of these onnx models.
 		2024-06-03 21:57:44 +05:30
..
AbstractInterpLibrary.cpp [Torch] Emit rrelu and decompose it (#3250) 2024-06-03 19:25:52 +08:00
AdjustCallingConventions.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
CMakeLists.txt [torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055) 2024-03-26 12:41:40 -07:00
DecomposeComplexOps.cpp [Torch] Emit rrelu and decompose it (#3250) 2024-06-03 19:25:52 +08:00
DropAbstractInterpCalculations.cpp Update to LLVM 029313cc979ae71877b65794b1063d4e51184cc8 2023-03-21 04:16:20 -07:00
EraseModuleInitializer.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
FuseQuantizedOps.cpp [TorchToLinalg] add support for quantized group conv (#3341) 2024-06-03 21:57:44 +05:30
GlobalizeObjectGraph.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
InlineGlobalSlots.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
LowerToBackendContract.cpp [Torch] Emit rrelu and decompose it (#3250) 2024-06-03 19:25:52 +08:00
MatchQuantizedOps.cpp Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
MaximizeValueSemantics.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
PassDetail.h llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Passes.cpp [Pipeline] Use dedicated simplification pipeline for TorchDynamo frontend (#3376) 2024-05-22 05:23:18 -07:00
PrepareForGlobalizeObjectGraph.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
RecomposeComplexOps.cpp [Torch] support recompose of aten.split.with_sizes and aten.tensor_sp… (#3401) 2024-05-31 09:56:47 +08:00
ReduceOpVariants.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
RefinePublicReturn.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
ReifyAbstractInterpCalculationsUtils.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
ReifyAbstractInterpCalculationsUtils.h handles 2,3,4 from https://github.com/llvm/torch-mlir/issues/1963 (#1964) 2023-03-24 21:50:01 -05:00
ReifyDtypeCalculations.cpp Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
ReifyShapeCalculations.cpp Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
ScalarizeShapes.cpp [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
SimplifyAbstractInterpCalculationsUtils.cpp Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
SimplifyAbstractInterpCalculationsUtils.h Replace RefineTypes with dtype functions (#2105) 2023-05-12 13:40:45 -07:00
SimplifyDtypeCalculations.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
SimplifyShapeCalculations.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00