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

remove fluid generator module (#50114)

上级 689de12c
......@@ -101,9 +101,7 @@ from .dygraph.layers import *
from .dygraph.base import enable_dygraph, disable_dygraph
from .io import save, load, load_program_state, set_program_state
from .dygraph.varbase_patch_methods import monkey_patch_varbase
from . import generator
from .core import _cuda_synchronize
from .generator import Generator
from .trainer_desc import (
TrainerDesc,
DistMultiTrainer,
......@@ -129,7 +127,6 @@ __all__ = (
+ data_feed_desc.__all__
+ compiler.__all__
+ backward.__all__
+ generator.__all__
+ [
'io',
'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
import paddle
import paddle.fluid as fluid
import paddle.fluid.core as core
import paddle.fluid.generator as generator
from paddle.tensor import random
......@@ -193,8 +192,6 @@ class TestGeneratorSeed(unittest.TestCase):
def test_generator_randint_dygraph(self):
"""Test Generator seed."""
gen = generator.Generator()
fluid.enable_dygraph()
gen = paddle.seed(12312321111)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册