diff options
| author | Florian Fischer <florian.fl.fischer@fau.de> | 2020-01-30 19:14:42 +0100 |
|---|---|---|
| committer | Florian Fischer <florian.fl.fischer@fau.de> | 2020-01-30 19:20:44 +0100 |
| commit | 08f920351884a4dfaffa4eb7423cce04374b9958 (patch) | |
| tree | f01900a9928e5e7d9c11e6dabac15da57a0cf958 | |
| parent | 0c85f85904010e8ddf9653d16857a94ec40c1e7f (diff) | |
| download | allocbench-08f920351884a4dfaffa4eb7423cce04374b9958.tar.gz allocbench-08f920351884a4dfaffa4eb7423cce04374b9958.zip | |
search for alloc_foo in src/allocators/alloc.py
| -rw-r--r-- | src/allocator.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/allocator.py b/src/allocator.py index 7b76543..fe31ed4 100644 --- a/src/allocator.py +++ b/src/allocator.py @@ -225,8 +225,8 @@ def collect_allocators(allocators): ret.update(read_allocators_collection_file(name)) # file is one of our allocator definitions import it - elif os.path.isfile("src/allocators/" + name + ".py"): - module = importlib.import_module('src.allocators.' + name) + elif os.path.isfile("src/allocators/" + name.split('_')[0] + ".py"): + module = importlib.import_module('src.allocators.' + name.split('_')[0]) # name is collection if hasattr(module, "allocators"): for alloc in module.allocators: |