Skip to content

Commit

Permalink
enable index_put test (#89019)
Browse files Browse the repository at this point in the history
Per title

Pull Request resolved: #89019
Approved by: https://github.com/desertfire
  • Loading branch information
ngimel authored and pytorchmergebot committed Nov 15, 2022
1 parent 68fd8f3 commit 5ed90c4
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions test/inductor/test_torchinductor_opinfo.py
Expand Up @@ -168,6 +168,8 @@ def process(device_type):
"__getitem__": {b8, f16, f32, f64, i32, i64},
"addr": {f16},
"allclose": {f16, f32, f64},
"amax": {f16},
"amin": {f16},
"angle": {f16, f32, f64},
"argwhere": {b8, f16, f32, f64, i32, i64},
"bernoulli": {f32, f64},
Expand Down Expand Up @@ -204,7 +206,6 @@ def process(device_type):
"fft.rfft2": {f32, f64},
"fft.rfftn": {f32, f64},
"index_add": {f16},
"index_put": {f16, f32, f64},
"index_reduce": {f16, f32, f64},
"istft": {f32, f64},
"linalg.eig": {f32, f64},
Expand Down Expand Up @@ -311,7 +312,6 @@ def process(device_type):
"fft.rfft": {f16, f32, f64},
"fft.rfft2": {f16, f32, f64},
"fft.rfftn": {f16, f32, f64},
"index_put": {f16, f32, f64},
"index_reduce": {f16, f32, f64},
"istft": {f32, f64},
"linalg.eig": {f32, f64},
Expand Down Expand Up @@ -441,13 +441,15 @@ def wrapper_set_seed(op, *args, **kwargs):
inductor_all_samples = {
"softmax.with_dtype",
"index_add",
"index_put",
"index_copy",
"scatter_reduce.sum",
"select_scatter",
"squeeze",
"unsqueeze",
"sum",
"amax",
"amin",
"all",
}


Expand Down Expand Up @@ -549,7 +551,6 @@ def fn(*args, **kwargs):
"check_gradient": requires_grad,
}
adjusted_kwargs.update(overridden_kwargs)

self.check_model_cuda(
fn,
args,
Expand Down

0 comments on commit 5ed90c4

Please sign in to comment.