Mentions légales du service

Skip to content
Snippets Groups Projects
Commit c66cc996 authored by hhakim's avatar hhakim
Browse files

Minor change: add a warning when pyfaust hierarchical2020/palm4msa2020 are gpu enabled.

parent fd703109
Branches
No related tags found
No related merge requests found
...@@ -460,6 +460,8 @@ def palm4msa(M, p, ret_lambda=False, backend=2016, on_gpu=False): ...@@ -460,6 +460,8 @@ def palm4msa(M, p, ret_lambda=False, backend=2016, on_gpu=False):
if on_gpu: raise ValueError("on_gpu applies only on 2020 backend.") if on_gpu: raise ValueError("on_gpu applies only on 2020 backend.")
core_obj, _lambda = _FaustCorePy.FaustFact.fact_palm4msa(M, p) core_obj, _lambda = _FaustCorePy.FaustFact.fact_palm4msa(M, p)
elif(backend == 2020): elif(backend == 2020):
if on_gpu: warnings.warn("on_gpu is totally experimental, use at your"
" own risk.")
core_obj, _lambda = _FaustCorePy.FaustFact.palm4msa2020(M, p, on_gpu) core_obj, _lambda = _FaustCorePy.FaustFact.palm4msa2020(M, p, on_gpu)
else: else:
raise ValueError("Unknown backend (only 2016 and 2020 are available).") raise ValueError("Unknown backend (only 2016 and 2020 are available).")
...@@ -644,6 +646,8 @@ def hierarchical(M, p, ret_lambda=False, ret_params=False, backend=2016, ...@@ -644,6 +646,8 @@ def hierarchical(M, p, ret_lambda=False, ret_params=False, backend=2016,
if on_gpu: raise ValueError("on_gpu applies only on 2020 backend.") if on_gpu: raise ValueError("on_gpu applies only on 2020 backend.")
core_obj,_lambda = _FaustCorePy.FaustFact.fact_hierarchical(M, p) core_obj,_lambda = _FaustCorePy.FaustFact.fact_hierarchical(M, p)
elif(backend == 2020): elif(backend == 2020):
if on_gpu: warnings.warn("on_gpu is totally experimental, use at your"
" own risk.")
core_obj, _lambda = _FaustCorePy.FaustFact.hierarchical2020(M, p, core_obj, _lambda = _FaustCorePy.FaustFact.hierarchical2020(M, p,
on_gpu) on_gpu)
else: else:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment