mirror of https://github.com/llvm/torch-mlir
parent
5253282c55
commit
c5d8c12469
|
@ -98,7 +98,7 @@ def sparse_export(
|
|||
annotation sparse parameters with their actual sparse layout
|
||||
attributes. This temporary solution accelerates testing
|
||||
torch-mlir with PyTorch sparse tensors until the issue is
|
||||
resovled.
|
||||
resolved.
|
||||
"""
|
||||
# Convert all arguments to dense.
|
||||
dargs = tuple(a.to_dense() if a.layout in SPARSE_LAYOUTS else a for a in args)
|
||||
|
|
Loading…
Reference in New Issue