lisyarus/pslang
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Aarch64 jit compiler wip: floating-point arithmetic operators

Browse source
This commit is contained in:
Nikita Lisitsa 2026-01-05 01:03:52 +03:00
parent c253041068
commit 8f5744c3f4
4 changed files with 76 additions and 65 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
examples/jit_test.psl
View file

@ -1,2 +1,3 @@
func test() -> f16: func test() -> f32:
return - 3.1415h let pi = 3.14159265358979323846
return pi / 2.0

5
libs/jit/include/pslang/jit/arch/aarch64/instruction_builder.hpp
View file

@ -140,6 +140,11 @@ namespace pslang::jit::aarch64
// Negate the floating-point register @reg_src and store the result in @reg_dst // Negate the floating-point register @reg_src and store the result in @reg_dst
void fneg(std::uint8_t reg_src, std::uint8_t mode, std::uint8_t reg_dst); void fneg(std::uint8_t reg_src, std::uint8_t mode, std::uint8_t reg_dst);
void fadd(std::uint8_t reg_src1, std::uint8_t reg_src2, std::uint8_t mode, std::uint8_t reg_dst);
void fsub(std::uint8_t reg_src1, std::uint8_t reg_src2, std::uint8_t mode, std::uint8_t reg_dst);
void fmul(std::uint8_t reg_src1, std::uint8_t reg_src2, std::uint8_t mode, std::uint8_t reg_dst);
void fdiv(std::uint8_t reg_src1, std::uint8_t reg_src2, std::uint8_t mode, std::uint8_t reg_dst);
// Return from a subroutine, taking the return address from register @reg // Return from a subroutine, taking the return address from register @reg
void ret(std::uint8_t reg = 30); void ret(std::uint8_t reg = 30);

91
libs/jit/source/arch/aarch64/compiler.cpp
View file

@ -361,89 +361,86 @@ namespace pslang::jit::aarch64
void apply(ast::binary_operation const & node) void apply(ast::binary_operation const & node)
{ {
// TODO: floating-point bool const is_fp = types::is_floating_point_type(*node.inferred_type);
std::uint8_t const fp_mode = fp_mode_for(*node.inferred_type);
if (is_fp)
{
apply(*node.arg1);
push_fp(0, fp_mode);
apply(*node.arg2);
pop_fp(1, fp_mode);
}
else
{
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
}
switch (node.type) switch (node.type)
{ {
case ast::binary_operation_type::addition: case ast::binary_operation_type::addition:
apply(*node.arg1); if (is_fp)
push(0); builder.fadd(1, 0, fp_mode, 0);
apply(*node.arg2); else
pop(1); {
builder.add_reg(1, 0, 0); builder.add_reg(1, 0, 0);
extend(0, node.inferred_type); extend(0, node.inferred_type);
}
break; break;
case ast::binary_operation_type::subtraction: case ast::binary_operation_type::subtraction:
apply(*node.arg1); if (is_fp)
push(0); builder.fsub(1, 0, fp_mode, 0);
apply(*node.arg2); else
pop(1); {
builder.sub_reg(1, 0, 0); builder.sub_reg(1, 0, 0);
extend(0, node.inferred_type); extend(0, node.inferred_type);
}
break; break;
case ast::binary_operation_type::multiplication: case ast::binary_operation_type::multiplication:
apply(*node.arg1); if (is_fp)
push(0); builder.fmul(1, 0, fp_mode, 0);
apply(*node.arg2); else
pop(1); {
builder.mul_reg(1, 0, 0); builder.mul_reg(1, 0, 0);
extend(0, node.inferred_type); extend(0, node.inferred_type);
}
break; break;
case ast::binary_operation_type::division: case ast::binary_operation_type::division:
apply(*node.arg1); if (is_fp)
push(0); builder.fdiv(1, 0, fp_mode, 0);
apply(*node.arg2); else
pop(1); {
if (types::is_signed_integer_type(*node.inferred_type)) if (types::is_signed_integer_type(*node.inferred_type))
builder.sdiv_reg(1, 0, 0); builder.sdiv_reg(1, 0, 0);
else else
builder.udiv_reg(1, 0, 0); builder.udiv_reg(1, 0, 0);
extend(0, node.inferred_type); extend(0, node.inferred_type);
}
break; break;
case ast::binary_operation_type::remainder: case ast::binary_operation_type::remainder:
// TODO: implement via div & mul & sub // TODO: implement via div & mul & sub
throw std::runtime_error("Not implemented"); throw std::runtime_error("Not implemented");
case ast::binary_operation_type::logical_and: case ast::binary_operation_type::logical_and:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.and_reg(1, 0, 0); builder.and_reg(1, 0, 0);
break; break;
case ast::binary_operation_type::logical_or: case ast::binary_operation_type::logical_or:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.or_reg(1, 0, 0); builder.or_reg(1, 0, 0);
break; break;
case ast::binary_operation_type::logical_xor: case ast::binary_operation_type::logical_xor:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.xor_reg(1, 0, 0); builder.xor_reg(1, 0, 0);
break; break;
case ast::binary_operation_type::equals: case ast::binary_operation_type::equals:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.cmp_reg(1, 0); builder.cmp_reg(1, 0);
builder.csetm(0, 0b0000); builder.csetm(0, 0b0000);
break; break;
case ast::binary_operation_type::not_equals: case ast::binary_operation_type::not_equals:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.cmp_reg(1, 0); builder.cmp_reg(1, 0);
builder.csetm(0, 0b0001); builder.csetm(0, 0b0001);
break; break;
case ast::binary_operation_type::less: case ast::binary_operation_type::less:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.cmp_reg(0, 1); builder.cmp_reg(0, 1);
if (types::is_bool_type(*ast::get_type(*node.arg1)) || types::is_unsigned_integer_type(*ast::get_type(*node.arg1))) if (types::is_bool_type(*ast::get_type(*node.arg1)) || types::is_unsigned_integer_type(*ast::get_type(*node.arg1)))
builder.csetm(0, 0b1000); builder.csetm(0, 0b1000);
@ -451,10 +448,6 @@ namespace pslang::jit::aarch64
builder.csetm(0, 0b1100); builder.csetm(0, 0b1100);
break; break;
case ast::binary_operation_type::greater: case ast::binary_operation_type::greater:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.cmp_reg(1, 0); builder.cmp_reg(1, 0);
if (types::is_bool_type(*ast::get_type(*node.arg1)) || types::is_unsigned_integer_type(*ast::get_type(*node.arg1))) if (types::is_bool_type(*ast::get_type(*node.arg1)) || types::is_unsigned_integer_type(*ast::get_type(*node.arg1)))
builder.csetm(0, 0b1000); builder.csetm(0, 0b1000);
@ -462,10 +455,6 @@ namespace pslang::jit::aarch64
builder.csetm(0, 0b1100); builder.csetm(0, 0b1100);
break; break;
case ast::binary_operation_type::less_equals: case ast::binary_operation_type::less_equals:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.cmp_reg(1, 0); builder.cmp_reg(1, 0);
if (types::is_bool_type(*ast::get_type(*node.arg1)) || types::is_unsigned_integer_type(*ast::get_type(*node.arg1))) if (types::is_bool_type(*ast::get_type(*node.arg1)) || types::is_unsigned_integer_type(*ast::get_type(*node.arg1)))
builder.csetm(0, 0b1001); builder.csetm(0, 0b1001);
@ -473,10 +462,6 @@ namespace pslang::jit::aarch64
builder.csetm(0, 0b1101); builder.csetm(0, 0b1101);
break; break;
case ast::binary_operation_type::greater_equals: case ast::binary_operation_type::greater_equals:
apply(*node.arg1);
push(0);
apply(*node.arg2);
pop(1);
builder.cmp_reg(0, 1); builder.cmp_reg(0, 1);
if (types::is_bool_type(*ast::get_type(*node.arg1)) || types::is_unsigned_integer_type(*ast::get_type(*node.arg1))) if (types::is_bool_type(*ast::get_type(*node.arg1)) || types::is_unsigned_integer_type(*ast::get_type(*node.arg1)))
builder.csetm(0, 0b1001); builder.csetm(0, 0b1001);

20
libs/jit/source/arch/aarch64/instruction_builder.cpp
View file

@ -184,6 +184,26 @@ namespace pslang::jit::aarch64
do_push(0x1e214000u | (reg_dst & REG_MASK) | ((reg_src & REG_MASK) << 5) | ((mode & 0x1u) << 22)); do_push(0x1e214000u | (reg_dst & REG_MASK) | ((reg_src & REG_MASK) << 5) | ((mode & 0x1u) << 22));
} }
void instruction_builder::fadd(std::uint8_t reg_src1, std::uint8_t reg_src2, std::uint8_t mode, std::uint8_t reg_dst)
{
do_push(0x1e202800u | (reg_dst & REG_MASK) | ((reg_src1 & REG_MASK) << 5) | ((reg_src2 & REG_MASK) << 16) | (((mode & 0x3u) ^ 0x2u) << 22));
}
void instruction_builder::fsub(std::uint8_t reg_src1, std::uint8_t reg_src2, std::uint8_t mode, std::uint8_t reg_dst)
{
do_push(0x1e203800u | (reg_dst & REG_MASK) | ((reg_src1 & REG_MASK) << 5) | ((reg_src2 & REG_MASK) << 16) | (((mode & 0x3u) ^ 0x2u) << 22));
}
void instruction_builder::fmul(std::uint8_t reg_src1, std::uint8_t reg_src2, std::uint8_t mode, std::uint8_t reg_dst)
{
do_push(0x1e200800u | (reg_dst & REG_MASK) | ((reg_src1 & REG_MASK) << 5) | ((reg_src2 & REG_MASK) << 16) | (((mode & 0x3u) ^ 0x2u) << 22));
}
void instruction_builder::fdiv(std::uint8_t reg_src1, std::uint8_t reg_src2, std::uint8_t mode, std::uint8_t reg_dst)
{
do_push(0x1e201800u | (reg_dst & REG_MASK) | ((reg_src1 & REG_MASK) << 5) | ((reg_src2 & REG_MASK) << 16) | (((mode & 0x3u) ^ 0x2u) << 22));
}
void instruction_builder::ret(std::uint8_t reg) void instruction_builder::ret(std::uint8_t reg)
{ {
do_push(0xd65f0000u | ((reg & REG_MASK) << 5)); do_push(0xd65f0000u | ((reg & REG_MASK) << 5));