https://github.com/shader-slang/slang
Tip revision: e59516fa8c3a16eb7b99a928c5b85b97bf44fd72 authored by Yong He on 01 February 2022, 00:26:03 UTC
Revise entrypoint renaming interface. (#2113)
Revise entrypoint renaming interface. (#2113)
Tip revision: e59516f
slang-ir-deduplicate.cpp
#include "slang-ir-insts.h"
namespace Slang
{
struct DeduplicateContext
{
SharedIRBuilder* builder;
IRInst* addValue(IRInst* value)
{
if (!value) return nullptr;
if (as<IRType>(value))
return addTypeValue(value);
if (auto constValue = as<IRConstant>(value))
return addConstantValue(constValue);
return value;
}
IRInst* addConstantValue(IRConstant* value)
{
IRConstantKey key = { value };
value->setFullType((IRType*)addValue(value->getFullType()));
if (auto newValue = builder->getConstantMap().TryGetValue(key))
return *newValue;
builder->getConstantMap()[key] = value;
return value;
}
IRInst* addTypeValue(IRInst* value)
{
// Do not deduplicate struct or interface types.
switch (value->getOp())
{
case kIROp_StructType:
case kIROp_InterfaceType:
return value;
default:
break;
}
for (UInt i = 0; i < value->getOperandCount(); i++)
{
value->setOperand(i, addValue(value->getOperand(i)));
}
value->setFullType((IRType*)addValue(value->getFullType()));
IRInstKey key = { value };
if (auto newValue = builder->getGlobalValueNumberingMap().TryGetValue(key))
return *newValue;
builder->getGlobalValueNumberingMap()[key] = value;
return value;
}
};
void SharedIRBuilder::deduplicateAndRebuildGlobalNumberingMap()
{
DeduplicateContext context;
context.builder = this;
bool changed = true;
m_constantMap.Clear();
m_globalValueNumberingMap.Clear();
for (auto inst : m_module->getGlobalInsts())
{
if (auto constVal = as<IRConstant>(inst))
{
context.addConstantValue(constVal);
}
}
List<IRInst*> instToRemove;
for (auto inst : m_module->getGlobalInsts())
{
if (as<IRType>(inst))
{
auto newInst = context.addTypeValue(inst);
if (newInst != inst)
{
changed = true;
inst->replaceUsesWith(newInst);
instToRemove.add(inst);
}
}
}
for (auto inst : instToRemove)
inst->removeAndDeallocate();
}
}