Ask Your Question
1

getting RESetMapReduce to use multiple cores

asked 2020-03-09 22:05:58 -0600

grobber gravatar image

updated 2020-03-10 08:53:08 -0600

I am reading the documentation on how to parallelize recursive-enumeration computations, and I cannot for the life of me get RESetMapReduce to use more than one core of my machine.

I am using the example immediately before the bulleted item titled Generating series on that page, with a minor modification:

sage: from sage.parallel.map_reduce import RESetMapReduce
sage: S = RESetMapReduce(
....:     roots=[[]],
....:     children=lambda l: [l + [0], l + [1]] if len(l) < 32 else [],
....:     map_function=lambda x: 1,
....:     reduce_function=lambda x, y: x + y,
....:     reduce_init=0)
sage: S.run()

I changed the length to 32 to make the computation heftier. If I run this and watch the process with htop I can see it using only one of my 8 cores to 100% capacity, ignoring the others.

I have even tried passing the argument max_proc=8 to the .run() method, to no effect.

edit retag flag offensive close merge delete

1 answer

Sort by » oldest newest most voted
1

answered 2020-03-09 22:16:48 -0600

grobber gravatar image

updated 2020-03-10 06:00:04 -0600

Actually, this worked: I ran the same computation in a terminal after setting the environment variable SAGE_NUM_THREADS to whatever was desired (8 in my case).

All cores are used in that case.

edit flag offensive delete link more

Comments

1

One may also

import os
os.environ["SAGE_NUM_THREADS"] = '16'

as suggested here: https://ask.sagemath.org/question/46711

Sébastien gravatar imageSébastien ( 2020-03-10 02:40:14 -0600 )edit

Thanks! I should perhaps have searched more thoroughly; this thread parallels that other one..

grobber gravatar imagegrobber ( 2020-03-10 06:00:48 -0600 )edit
2

It is a good thing to add to the documentation since this question often comes since ncpus() does not return the number of ncpus anymore. I posted a branch at https://trac.sagemath.org/ticket/29308

Sébastien gravatar imageSébastien ( 2020-03-10 06:23:29 -0600 )edit

Absolutely, that would help! The environment variable is mentioned in passing (twice) on the page I linked, which tipped me off to try to set it, but it's not made clear there that one needs it..

grobber gravatar imagegrobber ( 2020-03-10 08:23:56 -0600 )edit

Oh, and by the way, I tried this just now: if you run this as a sage script (I name mine <blah>.sage, make them executable, and put the shebang line #!/usr/bin/env sage at the top), then you don't actually need to import os: it just works. Maybe sage imports it automatically? I don't know, but I know the explicit import isn't needed.

grobber gravatar imagegrobber ( 2020-03-10 08:49:08 -0600 )edit

Your Answer

Please start posting anonymously - your entry will be published after you log in or create a new account.

Add Answer

Question Tools

1 follower

Stats

Asked: 2020-03-09 22:05:58 -0600

Seen: 45 times

Last updated: Mar 10