diff options
| author | Aart Bik <ajcbik@google.com> | 2022-01-14 11:08:57 -0800 |
|---|---|---|
| committer | Aart Bik <ajcbik@google.com> | 2022-01-14 16:20:31 -0800 |
| commit | e383eaa647da3a3f45313d22d83b6d6b5cdc9fc1 (patch) | |
| tree | 260bb8aaf73aea83362ff6b218d354a03d2cb299 /lldb/source/Plugins/ScriptInterpreter/Python/PythonDataObjects.cpp | |
| parent | a10692c734faff9ef28bc725703a0eacea78eeca (diff) | |
[mlir][sparse] parameterize MTTKRP kernel
Rather than hardcoding all constants, we now use the input tensor to drive the
code setup. Of course, we still need to hardcode dim-2 of A and the final
verification in CHECK is input dependent, but overall this sets a slightly
better example of tensor setup in general.
Reviewed By: bixia
Differential Revision: https://reviews.llvm.org/D117349
Diffstat (limited to 'lldb/source/Plugins/ScriptInterpreter/Python/PythonDataObjects.cpp')
0 files changed, 0 insertions, 0 deletions