提交 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}): ...@@ -204,7 +204,6 @@ def {sized_fn_name}({random_fn_input_names}):
@numba_funcify.register(aer.PoissonRV) @numba_funcify.register(aer.PoissonRV)
@numba_funcify.register(aer.GeometricRV) @numba_funcify.register(aer.GeometricRV)
@numba_funcify.register(aer.HyperGeometricRV) @numba_funcify.register(aer.HyperGeometricRV)
@numba_funcify.register(aer.CauchyRV)
@numba_funcify.register(aer.WaldRV) @numba_funcify.register(aer.WaldRV)
@numba_funcify.register(aer.LaplaceRV) @numba_funcify.register(aer.LaplaceRV)
@numba_funcify.register(aer.BinomialRV) @numba_funcify.register(aer.BinomialRV)
...@@ -271,6 +270,14 @@ def {np_random_fn_name}({np_input_names}): ...@@ -271,6 +270,14 @@ def {np_random_fn_name}({np_input_names}):
return make_numba_random_fn(node, np_random_fn) 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) @numba_funcify.register(aer.HalfNormalRV)
def numba_funcify_HalfNormalRV(op, node, **kwargs): def numba_funcify_HalfNormalRV(op, node, **kwargs):
def body_fn(a, b): def body_fn(a, b):
......
...@@ -2902,7 +2902,7 @@ def test_shared(): ...@@ -2902,7 +2902,7 @@ def test_shared():
), ),
], ],
at.as_tensor([3, 2]), 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, aer.wald,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论