modified random_sample.py

xgqdut2016 · xgqdut2016 · commit 3f90f69870bb · 2024-11-20T16:08:30.000+08:00
diff --git a/operatorspy/tests/random_sample.py b/operatorspy/tests/random_sample.py
@@ -83,7 +83,8 @@ def test(lib, handle, torch_device, voc, random_val, topp, topk, temperature, x_
         f"Testing RandomSample on {torch_device} with voc:{voc} dtype:{x_dtype}"
     )
     
-    data = torch.rand((voc), dtype=x_dtype).to(torch_device)
+    data = torch.arange(start=1, end=voc + 1, dtype=x_dtype).to(torch_device) / voc
+    data = data[torch.randperm(voc)]
     if(topp > 0 and topk > 1):
         ans = random_sample(data.to("cpu"), random_val, topp, topk, voc, temperature, "cpu")
     else:

Original file line number	Diff line number	Diff line change
`@@ -83,7 +83,8 @@ def test(lib, handle, torch_device, voc, random_val, topp, topk, temperature, x_`
`83`	`83`	`f"Testing RandomSample on {torch_device} with voc:{voc} dtype:{x_dtype}"`
`84`	`84`	`)`
`85`	`85`
`86`		`- data = torch.rand((voc), dtype=x_dtype).to(torch_device)`
	`86`	`+ data = torch.arange(start=1, end=voc + 1, dtype=x_dtype).to(torch_device) / voc`
	`87`	`+ data = data[torch.randperm(voc)]`
`87`	`88`	`if(topp > 0 and topk > 1):`
`88`	`89`	`ans = random_sample(data.to("cpu"), random_val, topp, topk, voc, temperature, "cpu")`
`89`	`90`	`else:`