Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

modify test_imperative_using_non_zero_gpu from use two gpus to one gpu #27348

Merged
merged 4 commits into from
Sep 17, 2020
Merged

modify test_imperative_using_non_zero_gpu from use two gpus to one gpu #27348

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
f1d796e
add op_function_generator.exe retry in windows, test=develop
wanghuancoder Aug 24, 2020
5892a73
Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into…
wanghuancoder Sep 15, 2020
2fc519d
Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into…
wanghuancoder Sep 16, 2020
952be54
modify test_imperative_using_non_zero_gpu from use two gpus to one gp…
wanghuancoder Sep 16, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion python/paddle/fluid/tests/unittests/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -560,7 +560,7 @@ endif()
set_tests_properties(test_parallel_executor_test_while_train test_parallel_executor_mnist
test_parallel_executor_feed_persistable_var
test_buffer_shared_memory_reuse_pass_and_fuse_optimization_op_pass
test_data_norm_op test_imperative_using_non_zero_gpu
test_data_norm_op
test_dataloader_keep_order
test_dataloader_unkeep_order
test_parallel_executor_fetch_isolated_var
Expand Down
2 changes: 0 additions & 2 deletions python/paddle/fluid/tests/unittests/test_imperative_using_non_zero_gpu.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@
class TestImperativeUsingNonZeroGpu(unittest.TestCase):
def run_main(self, np_arr, place):
with guard(place):
embedding = Embedding(size=[10, 10])
var = to_variable(np_arr)
self.assertTrue(np.array_equal(np_arr, var.numpy()))

Expand All @@ -30,7 +29,6 @@ def test_non_zero_gpu(self):
return

np_arr = np.random.random([11, 13]).astype('float32')
self.run_main(np_arr, fluid.CUDAPlace(1))
self.run_main(np_arr, fluid.CUDAPlace(0))


Expand Down