Skip to content

Commit

Permalink
backend/ninja: use a two step process for dependency scanning
Browse files Browse the repository at this point in the history
This splits the scanner into two discrete steps, one that scans the
source files, and one that that reads in the dependency information and
produces a dyndep.

The scanner uses the JSON format from
https://www.open-std.org/jtc1/sc22/wg21/docs/papers/2022/p1689r5.html,
which is the same format the MSVC and Clang use for C++ modules
scanning. This will allow us to more easily move to using MSVC and
clang-scan-deps when possible.

As an added bonus, this correctly tracks dependencies across TU and
Target boundaries, unlike the previous implementation, which assumed
that if it couldn't find a provider that everything was good, but could
run into issues. Because of that limitation Fortran code had to fully
depend on all of it's dependencies, transitive or not. Now, when using
the dep scanner, we can remove that restriction, allowing more
parallelism.
  • Loading branch information
dcbaker committed Mar 28, 2024
1 parent ce05b24 commit fd74510
Show file tree
Hide file tree
Showing 3 changed files with 254 additions and 66 deletions.
57 changes: 40 additions & 17 deletions mesonbuild/backend/ninjabackend.py
Original file line number Diff line number Diff line change
Expand Up @@ -994,7 +994,12 @@ def generate_target(self, target):
obj_targets = [t for t in od if t.uses_fortran()]
obj_list.extend(o)

fortran_order_deps = [File(True, *os.path.split(self.get_target_filename(t))) for t in obj_targets]
# We don't need this order dep if we're using dyndeps, as the
# depscanner will handle this for us, which produces a better dependency
# graph
fortran_order_deps: T.List[File] = []
if not self.use_dyndeps_for_fortran():
fortran_order_deps = [File(True, *os.path.split(self.get_target_filename(t))) for t in obj_targets]
fortran_inc_args: T.List[str] = []
if target.uses_fortran():
fortran_inc_args = mesonlib.listify([target.compilers['fortran'].get_include_args(
Expand Down Expand Up @@ -1102,7 +1107,7 @@ def generate_dependency_scan_target(self, target: build.BuildTarget,
if not self.should_use_dyndeps_for_target(target):
return
self._uses_dyndeps = True
depscan_file = self.get_dep_scan_file_for(target)
json_file, depscan_file = self.get_dep_scan_file_for(target)
pickle_base = target.name + '.dat'
pickle_file = os.path.join(self.get_target_private_dir(target), pickle_base).replace('\\', '/')
pickle_abs = os.path.join(self.get_target_private_dir_abs(target), pickle_base).replace('\\', '/')
Expand All @@ -1122,20 +1127,25 @@ def generate_dependency_scan_target(self, target: build.BuildTarget,
with open(pickle_abs, 'wb') as p:
pickle.dump(scaninfo, p)

elem = NinjaBuildElement(self.all_outputs, depscan_file, rule_name, pickle_file)
elem = NinjaBuildElement(self.all_outputs, json_file, rule_name, pickle_file)
# A full dependency is required on all scanned sources, if any of them
# are updated we need to rescan, as they may have changed the modules
# they use or export.
for s in scan_sources:
elem.deps.add(s[0])
# We need a full dependency on the output depfiles of other targets. If
# they change we need to completely
elem.orderdeps.update(object_deps)
elem.add_item('name', target.name)
self.add_build(elem)

infiles: T.Set[str] = set()
for t in target.get_transitive_link_deps():
if self.should_use_dyndeps_for_target(t):
elem.deps.add(os.path.join(self.get_target_dir(t), t.get_filename()))
elem.deps.update({os.path.join(self.get_target_dir(t), t.get_filename())
for t in self.flatten_object_list(target)[1]})
elem.orderdeps.update(object_deps)
infiles.add(self.get_dep_scan_file_for(t)[0])
_, od = self.flatten_object_list(target)
infiles.update({self.get_dep_scan_file_for(t)[0] for t in od if t.uses_fortran()})

elem = NinjaBuildElement(self.all_outputs, depscan_file, 'depaccumulate', [json_file] + sorted(infiles))
elem.add_item('name', target.name)
self.add_build(elem)

def select_sources_to_scan(self, compiled_sources: T.List[str],
Expand Down Expand Up @@ -2549,10 +2559,19 @@ def generate_scanner_rules(self):
if rulename in self.ruledict:
# Scanning command is the same for native and cross compilation.
return

command = self.environment.get_build_command() + \
['--internal', 'depscan']
args = ['$picklefile', '$out', '$in']
description = 'Module scanner.'
description = 'Scanning target $name for modules'
rule = NinjaRule(rulename, command, args, description)
self.add_rule(rule)

rulename = 'depaccumulate'
command = self.environment.get_build_command() + \
['--internal', 'depaccumulate']
args = ['$out', '$in']
description = 'Generating dynamic dependency information for target $name'
rule = NinjaRule(rulename, command, args, description)
self.add_rule(rule)

Expand Down Expand Up @@ -3058,8 +3077,9 @@ def generate_single_compile(self, target: build.BuildTarget, src,
d = os.path.join(self.get_target_private_dir(target), d)
element.add_orderdep(d)
element.add_dep(pch_dep)
for i in self.get_fortran_module_deps(target, compiler):
element.add_dep(i)
if not self.use_dyndeps_for_fortran():
for i in self.get_fortran_module_deps(target, compiler):
element.add_dep(i)
if dep_file:
element.add_item('DEPFILE', dep_file)
if compiler.get_language() == 'cuda':
Expand Down Expand Up @@ -3102,12 +3122,13 @@ def add_dependency_scanner_entries_to_element(self, target: build.BuildTarget, c
extension = extension.lower()
if not (extension in compilers.lang_suffixes['fortran'] or extension in compilers.lang_suffixes['cpp']):
return
dep_scan_file = self.get_dep_scan_file_for(target)
dep_scan_file = self.get_dep_scan_file_for(target)[1]
element.add_item('dyndep', dep_scan_file)
element.add_dep(dep_scan_file)

def get_dep_scan_file_for(self, target: build.BuildTarget) -> str:
return os.path.join(self.get_target_private_dir(target), 'depscan.dd')
def get_dep_scan_file_for(self, target: build.BuildTarget) -> T.Tuple[str, str]:
priv = self.get_target_private_dir(target)
return os.path.join(priv, 'depscan.json'), os.path.join(priv, 'depscan.dd')

def add_header_deps(self, target, ninja_element, header_deps):
for d in header_deps:
Expand All @@ -3131,8 +3152,10 @@ def has_dir_part(self, fname: mesonlib.FileOrString) -> bool:
# produce incorrect dep files but such is life. A full dependency is
# required to ensure that if a new module is added to an existing file that
# we correctly rebuild
def get_fortran_module_deps(self, target, compiler) -> T.List[str]:
if compiler.language != 'fortran':
def get_fortran_module_deps(self, target: build.BuildTarget, compiler: Compiler) -> T.List[str]:
# If we have dyndeps then we don't need this, since the depscanner will
# do all of things described above.
if compiler.language != 'fortran' or self.use_dyndeps_for_fortran():
return []
return [
os.path.join(self.get_target_dir(lt), lt.get_filename())
Expand Down
130 changes: 130 additions & 0 deletions mesonbuild/scripts/depaccumulate.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,130 @@
# SPDX-License-Identifier: Apache-2.0
# Copyright © 2021 Intel Corporation

"""Accumulator for p1689r5 module dependencies.
See: https://www.open-std.org/jtc1/sc22/wg21/docs/papers/2022/p1689r5.html
"""

from __future__ import annotations
import json
import re
import textwrap
import typing as T

if T.TYPE_CHECKING:
from .depscan import Description, Rule

# The quoting logic has been copied from the ninjabackend to avoid having to
# import half of Meson just to quote outputs, which is a performance problem
_QUOTE_PAT = re.compile(r"[$ :\n]")


def quote(text: str) -> str:
# Fast path for when no quoting is necessary
if not _QUOTE_PAT.search(text):
return text
if '\n' in text:
errmsg = textwrap.dedent(f'''\
Ninja does not support newlines in rules. The content was:
{text}
Please report this error with a test case to the Meson bug tracker.''')
raise RuntimeError(errmsg)
return _QUOTE_PAT.sub(r'$\g<0>', text)


_PROVIDER_CACHE: T.Dict[str, str] = {}


def get_provider(rules: T.List[Rule], name: str) -> T.Optional[str]:
"""Get the object that a module from another Target provides
We must rely on the object file here instead of the module itself, because
the object rule is part of the generated build.ninja, while the module is
only declared inside a dyndep. This creates for the dyndep generator to
depend on previous dyndeps as order deps. Since the module
interface file will be generated when the object is generated we can rely on
that in proxy and simplify generation.
:param rules: The list of rules to check
:param name: The logical-name to look for
:raises RuntimeError: If no provider can be found
:return: The object file of the rule providing the module
"""
# Cache the result for performance reasons
if name in _PROVIDER_CACHE:
return _PROVIDER_CACHE[name]

for r in rules:
for p in r.get('provides', []):
if p['logical-name'] == name:
obj = r['primary-output']
_PROVIDER_CACHE[name] = obj
return obj
return None


def process_rules(rules: T.List[Rule],
extra_rules: T.List[Rule],
) -> T.Iterable[T.Tuple[str, T.Optional[T.List[str]], T.Optional[T.List[str]]]]:
"""Process the rules for this Target
:param rules: the rules for this target
:param extra_rules: the rules for all of the targets this one links with, to use their provides
:yield: A tuple of the output, the exported modules, and the consumed modules
"""
for rule in rules:
prov: T.Optional[T.List[str]] = None
req: T.Optional[T.List[str]] = None
if 'provides' in rule:
prov = [p['compiled-module-path'] for p in rule['provides']]
if 'requires' in rule:
req = []
for p in rule['requires']:
modfile = p.get('compiled-module-path')
if modfile is not None:
req.append(modfile)
else:
# We can't error if this is not found because of compiler
# provided modules
found = get_provider(extra_rules, p['logical-name'])
if found:
req.append(found)
yield rule['primary-output'], prov, req


def format(files: T.Optional[T.List[str]]) -> str:
if files:
fmt = " ".join(quote(f) for f in files)
return f'| {fmt}'
return ''


def gen(outfile: str, desc: Description, extra_rules: T.List[Rule]) -> int:
with open(outfile, 'w', encoding='utf-8') as f:
f.write('ninja_dyndep_version = 1\n\n')

for obj, provides, requires in process_rules(desc['rules'], extra_rules):
ins = format(requires)
out = format(provides)
f.write(f'build {quote(obj)} {out}: dyndep {ins}\n\n')

return 0


def run(args: T.List[str]) -> int:
assert len(args) >= 2, 'got wrong number of arguments!'
outfile, jsonfile, *jsondeps = args
with open(jsonfile, 'r', encoding='utf-8') as f:
desc: Description = json.load(f)

# All rules, necessary for fulfilling across TU and target boundaries
rules = desc['rules'].copy()
for dep in jsondeps:
with open(dep, encoding='utf-8') as f:
d: Description = json.load(f)
rules.extend(d['rules'])

return gen(outfile, desc, rules)
Loading

0 comments on commit fd74510

Please sign in to comment.