Skip to content

Commit 6d6ee5b

Browse files
committed
Removed SIZEOF_* constants, renamings, convience lambda, feedback applied
1 parent d39a7cb commit 6d6ee5b

File tree

4 files changed

+167
-188
lines changed

4 files changed

+167
-188
lines changed

llvm/lib/Target/AMDGPU/AMDGPUAsmPrinter.cpp

Lines changed: 16 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -436,36 +436,34 @@ AMDGPUAsmPrinter::getAmdhsaKernelDescriptor(const MachineFunction &MF,
436436
const Function &F = MF.getFunction();
437437
const SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>();
438438

439+
auto CreateExpr = [&MF](int64_t Value) {
440+
return MCConstantExpr::create(Value, MF.getContext());
441+
};
442+
439443
MCKernelDescriptor KernelDescriptor;
440444

441445
assert(isUInt<32>(PI.ScratchSize));
442446
assert(isUInt<32>(PI.getComputePGMRSrc1(STM)));
443447
assert(isUInt<32>(PI.getComputePGMRSrc2()));
444448

445-
KernelDescriptor.group_segment_fixed_size =
446-
MCConstantExpr::create(PI.LDSSize, MF.getContext());
447-
KernelDescriptor.private_segment_fixed_size =
448-
MCConstantExpr::create(PI.ScratchSize, MF.getContext());
449+
KernelDescriptor.group_segment_fixed_size = CreateExpr(PI.LDSSize);
450+
KernelDescriptor.private_segment_fixed_size = CreateExpr(PI.ScratchSize);
449451

450452
Align MaxKernArgAlign;
451-
KernelDescriptor.kernarg_size = MCConstantExpr::create(
452-
STM.getKernArgSegmentSize(F, MaxKernArgAlign), MF.getContext());
453+
KernelDescriptor.kernarg_size =
454+
CreateExpr(STM.getKernArgSegmentSize(F, MaxKernArgAlign));
453455

454-
KernelDescriptor.compute_pgm_rsrc1 =
455-
MCConstantExpr::create(PI.getComputePGMRSrc1(STM), MF.getContext());
456-
KernelDescriptor.compute_pgm_rsrc2 =
457-
MCConstantExpr::create(PI.getComputePGMRSrc2(), MF.getContext());
458-
KernelDescriptor.kernel_code_properties = MCConstantExpr::create(
459-
getAmdhsaKernelCodeProperties(MF), MF.getContext());
456+
KernelDescriptor.compute_pgm_rsrc1 = CreateExpr(PI.getComputePGMRSrc1(STM));
457+
KernelDescriptor.compute_pgm_rsrc2 = CreateExpr(PI.getComputePGMRSrc2());
458+
KernelDescriptor.kernel_code_properties =
459+
CreateExpr(getAmdhsaKernelCodeProperties(MF));
460460

461461
assert(STM.hasGFX90AInsts() || CurrentProgramInfo.ComputePGMRSrc3GFX90A == 0);
462-
KernelDescriptor.compute_pgm_rsrc3 = MCConstantExpr::create(
463-
STM.hasGFX90AInsts() ? CurrentProgramInfo.ComputePGMRSrc3GFX90A : 0,
464-
MF.getContext());
462+
KernelDescriptor.compute_pgm_rsrc3 = CreateExpr(
463+
STM.hasGFX90AInsts() ? CurrentProgramInfo.ComputePGMRSrc3GFX90A : 0);
465464

466-
KernelDescriptor.kernarg_preload = MCConstantExpr::create(
467-
AMDGPU::hasKernargPreload(STM) ? Info->getNumKernargPreloadedSGPRs() : 0,
468-
MF.getContext());
465+
KernelDescriptor.kernarg_preload = CreateExpr(
466+
AMDGPU::hasKernargPreload(STM) ? Info->getNumKernargPreloadedSGPRs() : 0);
469467

470468
return KernelDescriptor;
471469
}

llvm/lib/Target/AMDGPU/AsmParser/AMDGPUAsmParser.cpp

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5488,15 +5488,17 @@ bool AMDGPUAsmParser::ParseDirectiveAMDHSAKernel() {
54885488
IDRange);
54895489

54905490
if (ID == ".amdhsa_group_segment_fixed_size") {
5491-
if (!isUInt<SIZEOF_GROUP_SEGMENT_FIXED_SIZE * CHAR_BIT>(Val))
5491+
if (!isUInt<sizeof(kernel_descriptor_t::group_segment_fixed_size) *
5492+
CHAR_BIT>(Val))
54925493
return OutOfRangeError(ValRange);
54935494
KD.group_segment_fixed_size = ExprVal;
54945495
} else if (ID == ".amdhsa_private_segment_fixed_size") {
5495-
if (!isUInt<SIZEOF_PRIVATE_SEGMENT_FIXED_SIZE * CHAR_BIT>(Val))
5496+
if (!isUInt<sizeof(kernel_descriptor_t::private_segment_fixed_size) *
5497+
CHAR_BIT>(Val))
54965498
return OutOfRangeError(ValRange);
54975499
KD.private_segment_fixed_size = ExprVal;
54985500
} else if (ID == ".amdhsa_kernarg_size") {
5499-
if (!isUInt<SIZEOF_KERNARG_SIZE * CHAR_BIT>(Val))
5501+
if (!isUInt<sizeof(kernel_descriptor_t::kernarg_size) * CHAR_BIT>(Val))
55005502
return OutOfRangeError(ValRange);
55015503
KD.kernarg_size = ExprVal;
55025504
} else if (ID == ".amdhsa_user_sgpr_count") {

llvm/lib/Target/AMDGPU/MCTargetDesc/AMDGPUMCKernelDescriptor.h

Lines changed: 0 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -39,29 +39,6 @@ struct MCKernelDescriptor {
3939
uint32_t Mask, MCContext &Ctx);
4040
};
4141

42-
enum : uint32_t {
43-
SIZEOF_GROUP_SEGMENT_FIXED_SIZE =
44-
sizeof(amdhsa::kernel_descriptor_t::group_segment_fixed_size),
45-
SIZEOF_PRIVATE_SEGMENT_FIXED_SIZE =
46-
sizeof(amdhsa::kernel_descriptor_t::private_segment_fixed_size),
47-
SIZEOF_KERNARG_SIZE = sizeof(amdhsa::kernel_descriptor_t::kernarg_size),
48-
SIZEOF_RESERVED0 = sizeof(amdhsa::kernel_descriptor_t::reserved0),
49-
SIZEOF_KERNEL_CODE_ENTRY_BYTE_OFFSET =
50-
sizeof(amdhsa::kernel_descriptor_t::kernel_code_entry_byte_offset),
51-
SIZEOF_RESERVED1 = sizeof(amdhsa::kernel_descriptor_t::reserved1),
52-
SIZEOF_COMPUTE_PGM_RSRC3 =
53-
sizeof(amdhsa::kernel_descriptor_t::compute_pgm_rsrc3),
54-
SIZEOF_COMPUTE_PGM_RSRC1 =
55-
sizeof(amdhsa::kernel_descriptor_t::compute_pgm_rsrc1),
56-
SIZEOF_COMPUTE_PGM_RSRC2 =
57-
sizeof(amdhsa::kernel_descriptor_t::compute_pgm_rsrc2),
58-
SIZEOF_KERNEL_CODE_PROPERTIES =
59-
sizeof(amdhsa::kernel_descriptor_t::kernel_code_properties),
60-
SIZEOF_KERNARG_PRELOAD = sizeof(amdhsa::kernel_descriptor_t::kernarg_preload),
61-
SIZEOF_RESERVED3 = sizeof(amdhsa::kernel_descriptor_t::reserved3),
62-
SIZEOF_KERNEL_DESCRIPTOR = sizeof(amdhsa::kernel_descriptor_t)
63-
};
64-
6542
} // end namespace AMDGPU
6643
} // end namespace llvm
6744

0 commit comments

Comments
 (0)