llvm/mlir/lib/ExecutionEngine/SparseTensor/MapRef.cpp

//===- MapRef.cpp - A dim2lvl/lvl2dim map reference wrapper ---------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "mlir/ExecutionEngine/SparseTensor/MapRef.h"
#include "mlir/Dialect/SparseTensor/IR/Enums.h"

mlir::sparse_tensor::MapRef::MapRef(uint64_t d, uint64_t l, const uint64_t *d2l,
                                    const uint64_t *l2d)
    : … { … }

bool mlir::sparse_tensor::MapRef::isPermutationMap() const { … }

bool mlir::sparse_tensor::MapRef::isFloor(uint64_t l, uint64_t &i,
                                          uint64_t &c) const { … }

bool mlir::sparse_tensor::MapRef::isMod(uint64_t l, uint64_t &i,
                                        uint64_t &c) const { … }

bool mlir::sparse_tensor::MapRef::isMul(uint64_t d, uint64_t &i, uint64_t &c,
                                        uint64_t &ii) const { … }