frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use random inputs for mps extension tests (#104597)

Browse Source 
The tested function simply adds `x` and `y`, given this fact, using random inputs instead of zeros makes more sense.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/104597
Approved by: https://github.com/albanD
This commit is contained in:
Li-Huai (Allan) Lin
2023-07-04 19:01:13 +08:00
committed by PyTorch MergeBot
parent 6bfd507c15
commit ac9c2aa6ee
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_cpp_extensions_aot.py
View File

@ -90,8 +90,8 @@ class TestCppExtensionAOT(common.TestCase):
import torch_test_cpp_extension.mps as mps_extension
tensor_length = 100000
x = torch.zeros(tensor_length, device="cpu", dtype=torch.float32)
y = torch.zeros(tensor_length, device="cpu", dtype=torch.float32)
x = torch.randn(tensor_length, device="cpu", dtype=torch.float32)
y = torch.randn(tensor_length, device="cpu", dtype=torch.float32)
cpu_output = mps_extension.get_cpu_add_output(x, y)
mps_output = mps_extension.get_mps_add_output(x.to("mps"), y.to("mps"))