b1703ad38d
A new test added for spirv-friendly builtins for SPV_KHR_bit_instructions unveiled that current mechanism to detect whether SPIRV Backend is in OpenCL environment or Vulkan environment was not good enough. This PR updates how to detect the environment and all the tests accordingly. *UPDATE*: the new approach is having a new member in `SPIRVSubtarget` to represent the environment. It can be either OpenCL, Kernel or Unknown. If the triple is explicit, we can directly set it at the creation of the `SPIRVSubtarget`, otherwise we just leave it unknown until we find other information that can help us set the environment. For now, the only other information we use to set the environment is `hlsl.shader` attribute at `SPIRV::ExecutionModel::ExecutionModel getExecutionModel(const SPIRVSubtarget &STI, const Function &F)`. Going forward we should consider also specific instructions that are Kernel-exclusive or Shader-exclusive. --------- Co-authored-by: marcos.maronas <mmaronas@smtp.igk.intel.com>
66 lines
2.5 KiB
C++
66 lines
2.5 KiB
C++
//===- SPIRVTargetTransformInfo.h - SPIR-V specific 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
// \file
|
|
// This file contains a TargetTransformInfoImplBase conforming object specific
|
|
// to the SPIRV target machine. It uses the target's detailed information to
|
|
// provide more precise answers to certain TTI queries, while letting the
|
|
// target independent and default TTI implementations handle the rest.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_SPIRV_SPIRVTARGETTRANSFORMINFO_H
|
|
#define LLVM_LIB_TARGET_SPIRV_SPIRVTARGETTRANSFORMINFO_H
|
|
|
|
#include "SPIRV.h"
|
|
#include "SPIRVTargetMachine.h"
|
|
#include "llvm/Analysis/TargetTransformInfo.h"
|
|
#include "llvm/CodeGen/BasicTTIImpl.h"
|
|
|
|
namespace llvm {
|
|
class SPIRVTTIImpl : public BasicTTIImplBase<SPIRVTTIImpl> {
|
|
using BaseT = BasicTTIImplBase<SPIRVTTIImpl>;
|
|
using TTI = TargetTransformInfo;
|
|
|
|
friend BaseT;
|
|
|
|
const SPIRVSubtarget *ST;
|
|
const SPIRVTargetLowering *TLI;
|
|
|
|
const TargetSubtargetInfo *getST() const { return ST; }
|
|
const SPIRVTargetLowering *getTLI() const { return TLI; }
|
|
|
|
public:
|
|
explicit SPIRVTTIImpl(const SPIRVTargetMachine *TM, const Function &F)
|
|
: BaseT(TM, F.getDataLayout()), ST(TM->getSubtargetImpl(F)),
|
|
TLI(ST->getTargetLowering()) {}
|
|
|
|
TTI::PopcntSupportKind getPopcntSupport(unsigned TyWidth) const override {
|
|
// SPIR-V natively supports OpBitcount, per 3.53.14 in the spec, as such it
|
|
// is reasonable to assume the Op is fast / preferable to the expanded loop.
|
|
// Furthermore, this prevents information being lost if transforms are
|
|
// applied to SPIR-V before lowering to a concrete target.
|
|
if (!isPowerOf2_32(TyWidth) || TyWidth > 64)
|
|
return TTI::PSK_Software; // Arbitrary bit-width INT is not core SPIR-V.
|
|
return TTI::PSK_FastHardware;
|
|
}
|
|
|
|
unsigned getFlatAddressSpace() const override {
|
|
if (ST->isShader())
|
|
return 0;
|
|
// FIXME: Clang has 2 distinct address space maps. One where
|
|
// default=4=Generic, and one with default=0=Function. This depends on the
|
|
// environment. For OpenCL, we don't need to run the InferAddrSpace pass, so
|
|
// we can return -1, but we might want to fix this.
|
|
return -1;
|
|
}
|
|
};
|
|
|
|
} // namespace llvm
|
|
|
|
#endif // LLVM_LIB_TARGET_SPIRV_SPIRVTARGETTRANSFORMINFO_H
|