admin/llgo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cl/instr:refine register collect

Browse Source
This commit is contained in:
luoliwoshang
2025-08-21 19:57:49 +08:00
parent f6bc5ac538
commit 0faef117ca
1 changed files with 10 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
cl/instr.go
View File

@@ -97,18 +97,20 @@ func (p *context) asmFull(b llssa.Builder, args []ssa.Value) (ret llssa.Expr) {
referrers := registerMap.Referrers() referrers := registerMap.Referrers()
for _, r := range *referrers { for _, r := range *referrers {
switch r := r.(type) { switch r := r.(type) {
case *ssa.DebugRef, *ssa.Call:
// ignore
case *ssa.MapUpdate: case *ssa.MapUpdate:
if r.Block() != registerMap.Block() { if r.Block() != registerMap.Block() {
panic("asmFull: register value map must be created in the same basic block") panic("asmFull: register value map must be created in the same basic block")
} }
key, ok := constStr(r.Key) key, ok := constStr(r.Key)
if !ok { if !ok {
panic("asmFull: register key must be a string constant") panic("asmFull: register key must be a string constant")
} }
llvmValue := p.compileValue(b, r.Value.(*ssa.MakeInterface).X) llvmValue := p.compileValue(b, r.Value.(*ssa.MakeInterface).X)
registers[key] = llvmValue registers[key] = llvmValue
default:
panic("asmFull: don't know how to handle argument to inline assembly: " + r.String())
} }
} }
} }
@@ -135,24 +137,16 @@ func (p *context) asmFull(b llssa.Builder, args []ssa.Value) (ret llssa.Expr) {
panic("asmFull: register not found: " + name) panic("asmFull: register not found: " + name)
} }
if _, ok := registerNumbers[name]; !ok { if _, ok := registerNumbers[name]; !ok {
// Type checking based on Go types - similar to TinyGo's implementation // Type checking - only allow integer basic types
rawType := value.Type.RawType() rawType := value.Type.RawType()
switch typ := rawType.Underlying().(type) { if basic, ok := rawType.Underlying().(*types.Basic); ok && basic.Info()&types.IsInteger != 0 {
case *types.Basic: registerNumbers[name] = len(registerNumbers)
if typ.Info()&types.IsInteger != 0 { inputValues = append(inputValues, value)
registerNumbers[name] = len(registerNumbers) constraints = append(constraints, "r")
inputValues = append(inputValues, value) } else {
constraints = append(constraints, "r")
} else {
panic("asmFull: unsupported basic type in inline assembly for operand: " + name + ", only integer types are supported")
}
case *types.Pointer:
// Pointer operands support was dropped, following TinyGo 0.23 // Pointer operands support was dropped, following TinyGo 0.23
panic("asmFull: not support for pointer operands: " + name + ", only integer types are supported")
default:
panic("asmFull: unsupported type in inline assembly for operand: " + name + ", only integer types are supported") panic("asmFull: unsupported type in inline assembly for operand: " + name + ", only integer types are supported")
} }
} }
return fmt.Sprintf("${%v}", registerNumbers[name]) return fmt.Sprintf("${%v}", registerNumbers[name])
}) })