提交 5ea74e97 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Fix the Numba implementation of CauchyRV

上级 0c7720cf
......@@ -204,7 +204,6 @@ def {sized_fn_name}({random_fn_input_names}):
@numba_funcify.register(aer.PoissonRV)
@numba_funcify.register(aer.GeometricRV)
@numba_funcify.register(aer.HyperGeometricRV)
@numba_funcify.register(aer.CauchyRV)
@numba_funcify.register(aer.WaldRV)
@numba_funcify.register(aer.LaplaceRV)
@numba_funcify.register(aer.BinomialRV)
......@@ -271,6 +270,14 @@ def {np_random_fn_name}({np_input_names}):
return make_numba_random_fn(node, np_random_fn)
@numba_funcify.register(aer.CauchyRV)
def numba_funcify_CauchyRV(op, node, **kwargs):
def body_fn(loc, scale):
return f" return ({loc} + np.random.standard_cauchy()) / {scale}"
return create_numba_random_fn(op, node, body_fn)
@numba_funcify.register(aer.HalfNormalRV)
def numba_funcify_HalfNormalRV(op, node, **kwargs):
def body_fn(a, b):
......
......@@ -2902,7 +2902,7 @@ def test_shared():
),
],
at.as_tensor([3, 2]),
marks=pytest.mark.xfail(reason="Not implemented"),
marks=pytest.mark.xfail(reason="Numba and NumPy rng states do not match"),
),
(
aer.wald,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论