e67cbbef03
This exposes creating a CallSiteLoc with a callee & list of frames for callers. Follows the creation approach in C++ side where a list of frames may be provided. Reviewed By: ftynse Differential Revision: https://reviews.llvm.org/D111670
90 lines
2.2 KiB
Python
90 lines
2.2 KiB
Python
# RUN: %PYTHON %s | FileCheck %s
|
|
|
|
import gc
|
|
from mlir.ir import *
|
|
|
|
def run(f):
|
|
print("\nTEST:", f.__name__)
|
|
f()
|
|
gc.collect()
|
|
assert Context._get_live_count() == 0
|
|
|
|
|
|
# CHECK-LABEL: TEST: testUnknown
|
|
def testUnknown():
|
|
with Context() as ctx:
|
|
loc = Location.unknown()
|
|
assert loc.context is ctx
|
|
ctx = None
|
|
gc.collect()
|
|
# CHECK: unknown str: loc(unknown)
|
|
print("unknown str:", str(loc))
|
|
# CHECK: unknown repr: loc(unknown)
|
|
print("unknown repr:", repr(loc))
|
|
|
|
run(testUnknown)
|
|
|
|
|
|
# CHECK-LABEL: TEST: testFileLineCol
|
|
def testFileLineCol():
|
|
with Context() as ctx:
|
|
loc = Location.file("foo.txt", 123, 56)
|
|
ctx = None
|
|
gc.collect()
|
|
# CHECK: file str: loc("foo.txt":123:56)
|
|
print("file str:", str(loc))
|
|
# CHECK: file repr: loc("foo.txt":123:56)
|
|
print("file repr:", repr(loc))
|
|
|
|
run(testFileLineCol)
|
|
|
|
|
|
# CHECK-LABEL: TEST: testName
|
|
def testName():
|
|
with Context() as ctx:
|
|
loc = Location.name("nombre")
|
|
locWithChildLoc = Location.name("naam", loc)
|
|
ctx = None
|
|
gc.collect()
|
|
# CHECK: file str: loc("nombre")
|
|
print("file str:", str(loc))
|
|
# CHECK: file repr: loc("nombre")
|
|
print("file repr:", repr(loc))
|
|
# CHECK: file str: loc("naam"("nombre"))
|
|
print("file str:", str(locWithChildLoc))
|
|
# CHECK: file repr: loc("naam"("nombre"))
|
|
print("file repr:", repr(locWithChildLoc))
|
|
|
|
run(testName)
|
|
|
|
|
|
# CHECK-LABEL: TEST: testCallSite
|
|
def testCallSite():
|
|
with Context() as ctx:
|
|
loc = Location.callsite(
|
|
Location.file("foo.text", 123, 45), [
|
|
Location.file("util.foo", 379, 21),
|
|
Location.file("main.foo", 100, 63)
|
|
])
|
|
ctx = None
|
|
# CHECK: file str: loc(callsite("foo.text":123:45 at callsite("util.foo":379:21 at "main.foo":100:63))
|
|
print("file str:", str(loc))
|
|
# CHECK: file repr: loc(callsite("foo.text":123:45 at callsite("util.foo":379:21 at "main.foo":100:63))
|
|
print("file repr:", repr(loc))
|
|
|
|
run(testCallSite)
|
|
|
|
|
|
# CHECK-LABEL: TEST: testLocationCapsule
|
|
def testLocationCapsule():
|
|
with Context() as ctx:
|
|
loc1 = Location.file("foo.txt", 123, 56)
|
|
# CHECK: mlir.ir.Location._CAPIPtr
|
|
loc_capsule = loc1._CAPIPtr
|
|
print(loc_capsule)
|
|
loc2 = Location._CAPICreate(loc_capsule)
|
|
assert loc2 == loc1
|
|
assert loc2.context is ctx
|
|
|
|
run(testLocationCapsule)
|