d8f71b1638
Fixes the first bullet in #184638
Corresponding patch to add support for vector operands in
OpConvertPtrToU/OpConvertUToPtr operations in spirv-val:
https://github.com/KhronosGroup/SPIRV-Tools/pull/6575
SPIR-V extension reference used:
278044a51f/extensions/INTEL/SPV_INTEL_masked_gather_scatter.asciidoc
50 lines
1.7 KiB
C++
50 lines
1.7 KiB
C++
//===- SPIRVTargetTransformInfo.cpp - 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "SPIRVTargetTransformInfo.h"
|
|
#include "SPIRVSubtarget.h"
|
|
#include "llvm/IR/IntrinsicsSPIRV.h"
|
|
|
|
using namespace llvm;
|
|
|
|
bool llvm::SPIRVTTIImpl::collectFlatAddressOperands(
|
|
SmallVectorImpl<int> &OpIndexes, Intrinsic::ID IID) const {
|
|
switch (IID) {
|
|
case Intrinsic::spv_generic_cast_to_ptr_explicit:
|
|
OpIndexes.push_back(0);
|
|
return true;
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
Value *llvm::SPIRVTTIImpl::rewriteIntrinsicWithAddressSpace(IntrinsicInst *II,
|
|
Value *OldV,
|
|
Value *NewV) const {
|
|
auto IntrID = II->getIntrinsicID();
|
|
switch (IntrID) {
|
|
case Intrinsic::spv_generic_cast_to_ptr_explicit: {
|
|
unsigned NewAS = NewV->getType()->getPointerAddressSpace();
|
|
unsigned DstAS = II->getType()->getPointerAddressSpace();
|
|
return NewAS == DstAS ? NewV
|
|
: ConstantPointerNull::get(
|
|
PointerType::get(NewV->getContext(), DstAS));
|
|
}
|
|
default:
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
bool SPIRVTTIImpl::isLegalMaskedGather(Type *DataType, Align Alignment) const {
|
|
return ST->canUseExtension(SPIRV::Extension::SPV_INTEL_masked_gather_scatter);
|
|
}
|
|
|
|
bool SPIRVTTIImpl::isLegalMaskedScatter(Type *DataType, Align Alignment) const {
|
|
return ST->canUseExtension(SPIRV::Extension::SPV_INTEL_masked_gather_scatter);
|
|
}
|