Skip to content

Commit d2ee9e1

Browse files
committed
Fix LLVM execution
1 parent 675ef54 commit d2ee9e1

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

ydb/library/yql/minikql/comp_nodes/mkql_wide_top_sort.cpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -545,7 +545,6 @@ using TBase = TComputationValue<TSpillingSupportState<Sort, HasCount>>;
545545

546546
bool HasMemoryForProcessing() const {
547547
return !TlsAllocState->IsMemoryYellowZoneEnabled();
548-
// return false;
549548
}
550549

551550
bool IsReadFromChannelFinished() const {
@@ -632,9 +631,9 @@ using TBase = TComputationValue<TSpillingSupportState<Sort, HasCount>>;
632631

633632
#ifndef MKQL_DISABLE_CODEGEN
634633
template <bool Sort, bool HasCount>
635-
class TLLVMFieldsStructureState: public TLLVMFieldsStructure<TComputationValue<TSpillingSupportState<Sort, HasCount>>> {
634+
class TLLVMFieldsStructureState: public TLLVMFieldsStructure<TComputationValue<TState<Sort, HasCount>>> {
636635
private:
637-
using TBase = TLLVMFieldsStructure<TComputationValue<TSpillingSupportState<Sort, HasCount>>>;
636+
using TBase = TLLVMFieldsStructure<TComputationValue<TState<Sort, HasCount>>>;
638637
llvm::IntegerType* ValueType;
639638
llvm::PointerType* PtrValueType;
640639
llvm::IntegerType* StatusType;
@@ -841,7 +840,7 @@ using TBaseComputation = TStatefulWideFlowCodegeneratorNode<TWideTopWrapper<Sort
841840
block = rest;
842841

843842
new StoreInst(ConstantInt::get(last->getType(), static_cast<i32>(EFetchResult::Finish)), statusPtr, block);
844-
const auto sealFunc = ConstantInt::get(Type::getInt64Ty(context), GetMethodPtr(&TSpillingSupportState<Sort, HasCount>::Seal));
843+
const auto sealFunc = ConstantInt::get(Type::getInt64Ty(context), GetMethodPtr(&TState<Sort, HasCount>::Seal));
845844
const auto sealType = FunctionType::get(Type::getVoidTy(context), {stateArg->getType()}, false);
846845
const auto sealPtr = CastInst::Create(Instruction::IntToPtr, sealFunc, PointerType::getUnqual(sealType), "seal", block);
847846
CallInst::Create(sealType, sealPtr, {stateArg}, "", block);
@@ -872,7 +871,7 @@ using TBaseComputation = TStatefulWideFlowCodegeneratorNode<TWideTopWrapper<Sort
872871
}
873872

874873

875-
const auto pushFunc = ConstantInt::get(Type::getInt64Ty(context), GetMethodPtr(&TSpillingSupportState<Sort, HasCount>::Put));
874+
const auto pushFunc = ConstantInt::get(Type::getInt64Ty(context), GetMethodPtr(&TState<Sort, HasCount>::Put));
876875
const auto pushType = FunctionType::get(Type::getInt1Ty(context), {stateArg->getType()}, false);
877876
const auto pushPtr = CastInst::Create(Instruction::IntToPtr, pushFunc, PointerType::getUnqual(pushType), "function", block);
878877
const auto accepted = CallInst::Create(pushType, pushPtr, {stateArg}, "accepted", block);
@@ -912,7 +911,7 @@ using TBaseComputation = TStatefulWideFlowCodegeneratorNode<TWideTopWrapper<Sort
912911

913912
const auto good = BasicBlock::Create(context, "good", ctx.Func);
914913

915-
const auto extractFunc = ConstantInt::get(Type::getInt64Ty(context), GetMethodPtr(&TSpillingSupportState<Sort, HasCount>::Extract));
914+
const auto extractFunc = ConstantInt::get(Type::getInt64Ty(context), GetMethodPtr(&TState<Sort, HasCount>::Extract));
916915
const auto extractType = FunctionType::get(outputPtrType, {stateArg->getType()}, false);
917916
const auto extractPtr = CastInst::Create(Instruction::IntToPtr, extractFunc, PointerType::getUnqual(extractType), "extract", block);
918917
const auto out = CallInst::Create(extractType, extractPtr, {stateArg}, "out", block);

0 commit comments

Comments
 (0)