Skip to content

Commit

Permalink
[ROCm] Enable Lerp tests for complex32 (#108100)
Browse files Browse the repository at this point in the history
Enables previously disabled "lerp" opinfo tests for chalf on ROCm

Pull Request resolved: #108100
Approved by: https://github.com/pruthvistony, https://github.com/jithunnair-amd, https://github.com/kit1980
  • Loading branch information
alugorey authored and pytorchmergebot committed Sep 14, 2023
1 parent 54c5f47 commit ea94344
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion torch/testing/_internal/common_methods_invocations.py
Original file line number Diff line number Diff line change
Expand Up @@ -15244,7 +15244,6 @@ def reference_flatten(input, start_dim=0, end_dim=-1):
OpInfo('lerp',
dtypes=floating_and_complex_types_and(torch.bfloat16, torch.half),
dtypesIfCUDA=floating_and_complex_types_and(torch.chalf, torch.half, torch.bfloat16),
dtypesIfROCM=floating_and_complex_types_and(torch.half, torch.bfloat16),
sample_inputs_func=sample_inputs_lerp,
supports_forward_ad=True,
supports_fwgrad_bwgrad=True,
Expand Down

0 comments on commit ea94344

Please sign in to comment.