8663b8777e
This changes allows target intrinsics to specify and overwrite overloaded types. - Updates `ReplaceWithVecLib` to not provide TTI as there most probably won't be a use-case - Updates `SLPVectorizer` to use available TTI - Updates `VPTransformState` to pass down TTI - Updates `VPlanRecipe` to use passed-down TTI This change will let us add scalarization for `asdouble`: #114847
46 lines
1.7 KiB
C++
46 lines
1.7 KiB
C++
//===- DirectXTargetTransformInfo.h - DirectX TTI ---------------*- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_DIRECTX_DIRECTXTARGETTRANSFORMINFO_H
|
|
#define LLVM_DIRECTX_DIRECTXTARGETTRANSFORMINFO_H
|
|
|
|
#include "DirectXSubtarget.h"
|
|
#include "DirectXTargetMachine.h"
|
|
#include "llvm/CodeGen/BasicTTIImpl.h"
|
|
#include "llvm/IR/Function.h"
|
|
|
|
namespace llvm {
|
|
class DirectXTTIImpl : public BasicTTIImplBase<DirectXTTIImpl> {
|
|
using BaseT = BasicTTIImplBase<DirectXTTIImpl>;
|
|
using TTI = TargetTransformInfo;
|
|
|
|
friend BaseT;
|
|
|
|
const DirectXSubtarget *ST;
|
|
const DirectXTargetLowering *TLI;
|
|
|
|
const DirectXSubtarget *getST() const { return ST; }
|
|
const DirectXTargetLowering *getTLI() const { return TLI; }
|
|
|
|
public:
|
|
explicit DirectXTTIImpl(const DirectXTargetMachine *TM, const Function &F)
|
|
: BaseT(TM, F.getDataLayout()), ST(TM->getSubtargetImpl(F)),
|
|
TLI(ST->getTargetLowering()) {}
|
|
unsigned getMinVectorRegisterBitWidth() const { return 32; }
|
|
bool isTargetIntrinsicTriviallyScalarizable(Intrinsic::ID ID) const;
|
|
bool isTargetIntrinsicWithScalarOpAtArg(Intrinsic::ID ID,
|
|
unsigned ScalarOpdIdx);
|
|
bool isVectorIntrinsicWithOverloadTypeAtArg(Intrinsic::ID ID,
|
|
int ScalarOpdIdx);
|
|
};
|
|
} // namespace llvm
|
|
|
|
#endif // LLVM_DIRECTX_DIRECTXTARGETTRANSFORMINFO_H
|