060df78cdb
Completes #83626 - `CGBuiltin.cpp` - modify `getDotProductIntrinsic` to be able to emit `dot2`, `dot3`, and `dot4` intrinsics based on element count - `IntrinsicsDirectX.td` - for floating point add `dot2`, `dot3`, and `dot4` inntrinsics -`DXIL.td` add dxilop intrinsic lowering for `dot2`, `dot3`, & `dot4`. - `DXILOpLowering.cpp` - add vector arg flattening for dot product. - `DXILOpBuilder.h` - modify `createDXILOpCall` to take a smallVector instead of an iterator - `DXILOpBuilder.cpp` - modify `createDXILOpCall` by moving the small vector up to the calling function in `DXILOpLowering.cpp`. - Moving one function up gives us access to the `CallInst` and `Function` which were needed to distinguish the dot product intrinsics and get the operands without using the iterator.
51 lines
1.6 KiB
C++
51 lines
1.6 KiB
C++
//===- DXILOpBuilder.h - Helper class for build DIXLOp functions ----------===//
|
|
//
|
|
// 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 class to help build DXIL op functions.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_DIRECTX_DXILOPBUILDER_H
|
|
#define LLVM_LIB_TARGET_DIRECTX_DXILOPBUILDER_H
|
|
|
|
#include "DXILConstants.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
|
|
namespace llvm {
|
|
class Module;
|
|
class IRBuilderBase;
|
|
class CallInst;
|
|
class Value;
|
|
class Type;
|
|
class FunctionType;
|
|
class Use;
|
|
|
|
namespace dxil {
|
|
|
|
class DXILOpBuilder {
|
|
public:
|
|
DXILOpBuilder(Module &M, IRBuilderBase &B) : M(M), B(B) {}
|
|
/// Create an instruction that calls DXIL Op with return type, specified
|
|
/// opcode, and call arguments. \param OpCode Opcode of the DXIL Op call
|
|
/// constructed \param ReturnTy Return type of the DXIL Op call constructed
|
|
/// \param OverloadTy Overload type of the DXIL Op call constructed
|
|
/// \return DXIL Op call constructed
|
|
CallInst *createDXILOpCall(dxil::OpCode OpCode, Type *ReturnTy,
|
|
Type *OverloadTy, SmallVector<Value *> Args);
|
|
Type *getOverloadTy(dxil::OpCode OpCode, FunctionType *FT);
|
|
static const char *getOpCodeName(dxil::OpCode DXILOp);
|
|
|
|
private:
|
|
Module &M;
|
|
IRBuilderBase &B;
|
|
};
|
|
|
|
} // namespace dxil
|
|
} // namespace llvm
|
|
|
|
#endif
|