utils.select_module module

RosenPy: An Open Source Python Framework for Complex-Valued Neural Networks. Copyright © A. A. Cruz, K. S. Mayer, D. S. Arantes.

License:

This file is part of RosenPy. RosenPy is an open source framework distributed under the terms of the GNU General Public License, as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. For additional information on license terms, please open the Readme.md file.

RosenPy is distributed in the hope that it will be useful to every user, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with RosenPy. If not, see <http://www.gnu.org/licenses/>.

utils.select_module.select_module(use_gpu)[source]

This function is used to implement CPU/GPU generic code.

Parameters:

use_gpu (bool) – A flag indicating whether to use GPU acceleration (CuPy) or CPU (NumPy).

Returns:

module – Returns the module cupy if available and requested, otherwise numpy.

Return type:

module