1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
# Copyright (C) 2019 Chris Richardson
#
# This file is part of FFCx. (https://www.fenicsproject.org)
#
# SPDX-License-Identifier: LGPL-3.0-or-later
import sys
import ffcx.codegeneration.jit
import ufl
def test_cache_modes(compile_args):
cell = ufl.triangle
element = ufl.FiniteElement("Lagrange", cell, 1)
u, v = ufl.TrialFunction(element), ufl.TestFunction(element)
a = ufl.inner(ufl.grad(u), ufl.grad(v)) * ufl.dx
forms = [a]
# Load form from /tmp
compiled_forms, module, code = ffcx.codegeneration.jit.compile_forms(forms, cffi_extra_compile_args=compile_args)
tmpname = module.__name__
tmpfile = module.__file__
print(tmpname, tmpfile)
del sys.modules[tmpname]
# Load form from cache
compiled_forms, module, code = ffcx.codegeneration.jit.compile_forms(
forms, cache_dir="./compile-cache", cffi_extra_compile_args=compile_args)
newname = module.__name__
newfile = module.__file__
print(newname, newfile)
assert newname == tmpname
assert newfile != tmpfile
|
