未验证 提交 08caa3e7 编写于 作者: J JYChen 提交者: GitHub

remove fluid generator module (#50114)

上级 689de12c
...@@ -101,9 +101,7 @@ from .dygraph.layers import * ...@@ -101,9 +101,7 @@ from .dygraph.layers import *
from .dygraph.base import enable_dygraph, disable_dygraph from .dygraph.base import enable_dygraph, disable_dygraph
from .io import save, load, load_program_state, set_program_state from .io import save, load, load_program_state, set_program_state
from .dygraph.varbase_patch_methods import monkey_patch_varbase from .dygraph.varbase_patch_methods import monkey_patch_varbase
from . import generator
from .core import _cuda_synchronize from .core import _cuda_synchronize
from .generator import Generator
from .trainer_desc import ( from .trainer_desc import (
TrainerDesc, TrainerDesc,
DistMultiTrainer, DistMultiTrainer,
...@@ -129,7 +127,6 @@ __all__ = ( ...@@ -129,7 +127,6 @@ __all__ = (
+ data_feed_desc.__all__ + data_feed_desc.__all__
+ compiler.__all__ + compiler.__all__
+ backward.__all__ + backward.__all__
+ generator.__all__
+ [ + [
'io', 'io',
'initializer', 'initializer',
......
# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""This is definition of generator class, which is for managing the state of the algorithm that produces pseudo random numbers."""
from . import core
from .framework import _get_paddle_place
__all__ = ['Generator']
class Generator(core.Generator):
"""Generator class"""
def __init__(self, place=None):
"""
Create a generator object which manages the random number generation. ( Experimental Feature )
Parameters:
place(CPUPlace|CUDAPinnedPlace|CUDAPlace|str,optional): The place to allocate Tensor. Can be
CPUPlace, CUDAPinnedPlace, CUDAPlace. Default: None, means global place. If ``place`` is
string, it can be ``cpu`` and ``gpu:x``, where ``x`` is the index of the GPUs.
Returns:
Generator: A generator object.
"""
self.place = _get_paddle_place(place)
if not place:
place = core.CPUPlace()
if isinstance(place, core.CPUPlace):
super().__init__()
else:
raise ValueError(
"Generator class with %s does is not supported yet, currently only support generator with CPUPlace "
% place
)
# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Test cloud role maker."""
import unittest
import paddle
import paddle.fluid.generator as generator
class TestGenerator(unittest.TestCase):
"""
Test cases for cpu generator.
"""
def test_basic_generator(self):
"""Test basic generator."""
gen = generator.Generator()
gen.manual_seed(123123143)
st = gen.get_state()
gen.set_state(st)
gen.random()
def test_basic_generator_error(self):
if paddle.fluid.core.is_compiled_with_cuda():
self.assertRaises(
ValueError, generator.Generator, place=paddle.CUDAPlace(0)
)
if __name__ == "__main__":
unittest.main()
...@@ -20,7 +20,6 @@ import numpy as np ...@@ -20,7 +20,6 @@ import numpy as np
import paddle import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.fluid.core as core import paddle.fluid.core as core
import paddle.fluid.generator as generator
from paddle.tensor import random from paddle.tensor import random
...@@ -193,8 +192,6 @@ class TestGeneratorSeed(unittest.TestCase): ...@@ -193,8 +192,6 @@ class TestGeneratorSeed(unittest.TestCase):
def test_generator_randint_dygraph(self): def test_generator_randint_dygraph(self):
"""Test Generator seed.""" """Test Generator seed."""
gen = generator.Generator()
fluid.enable_dygraph() fluid.enable_dygraph()
gen = paddle.seed(12312321111) gen = paddle.seed(12312321111)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册