Feature(compiler): BinaryExpression - support pow (a ** 2)

ovr · Jan 11, 2019 · 13a1011 · 13a1011
1 parent 236ad40
commit 13a1011
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 0 deletions.
diff --git a/src/backend/llvm/code-generation/binary-expression.ts b/src/backend/llvm/code-generation/binary-expression.ts
@@ -31,6 +31,7 @@ export class BinaryExpressionCodeGenerator implements NodeGenerateInterface<ts.B
             case ts.SyntaxKind.LessThanLessThanEqualsToken:
             case ts.SyntaxKind.GreaterThanGreaterThanEqualsToken:
             case ts.SyntaxKind.CaretEqualsToken:
+            case ts.SyntaxKind.AsteriskAsteriskEqualsToken:
             case ts.SyntaxKind.AsteriskEqualsToken:
             case ts.SyntaxKind.PlusEqualsToken:
             case ts.SyntaxKind.MinusEqualsToken: {
@@ -101,6 +102,22 @@ export class BinaryExpressionCodeGenerator implements NodeGenerateInterface<ts.B
                     )
                 );
             }
+            // a ** b
+            case ts.SyntaxKind.AsteriskAsteriskEqualsToken:
+            case ts.SyntaxKind.AsteriskAsteriskToken: {
+                const left = buildFromExpression(node.left, ctx, builder);
+                const right = buildFromExpression(node.right, ctx, builder);
+
+                return new Primitive(
+                    builder.createCall(
+                        ctx.getIntrinsic('llvm.pow.f64'),
+                        [
+                            loadIfNeeded(left, builder),
+                            loadIfNeeded(right, builder)
+                        ]
+                    )
+                );
+            }
             // a ^ b
             case ts.SyntaxKind.CaretEqualsToken:
             case ts.SyntaxKind.CaretToken: {

diff --git a/src/backend/llvm/context.ts b/src/backend/llvm/context.ts
@@ -70,4 +70,32 @@ export class Context {
             'array<float64>'
         ));
     }
+
+    getIntrinsic(functionName: string): llvm.Function {
+        const moduleFn = this.llvmModule.getFunction(functionName);
+        if (moduleFn) {
+            return moduleFn;
+        }
+
+        switch (functionName) {
+            case 'llvm.pow.f64':
+                const intrinsicType = llvm.FunctionType.get(
+                    llvm.Type.getDoubleTy(this.llvmContext),
+                    [
+                        llvm.Type.getDoubleTy(this.llvmContext),
+                        llvm.Type.getDoubleTy(this.llvmContext),
+                    ],
+                    false
+                );
+
+                return llvm.Function.create(
+                    intrinsicType,
+                    llvm.LinkageTypes.ExternalLinkage,
+                    functionName,
+                    this.llvmModule
+                );
+            default:
+                throw new Error(`Unknown intrinsic: "${functionName}"`);
+        }
+    }
 }