Skip to content

Commit

Permalink
Enable core aten test and loose tolerance for aten_log10 and aten_log2 (
Browse files Browse the repository at this point in the history
  • Loading branch information
wonjoolee95 authored and mbzomowski committed Jan 3, 2024
1 parent f9092cc commit 73b7d7d
Showing 1 changed file with 16 additions and 7 deletions.
23 changes: 16 additions & 7 deletions test/test_core_aten_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -2353,19 +2353,23 @@ def test_aten_log_2(self):
kwargs = dict()
run_export_and_compare(self, torch.ops.aten.log, args, kwargs)

@unittest.skip
def test_aten_log10_0(self):
args = (torch.randn((10, 10)).to(torch.float32),)
kwargs = dict()
run_export_and_compare(self, torch.ops.aten.log10, args, kwargs)

@unittest.skip
def test_aten_log10_1(self):
args = (torch.randn((10, 10)).to(torch.float16),)
kwargs = dict()
run_export_and_compare(self, torch.ops.aten.log10, args, kwargs)
run_export_and_compare(
self,
torch.ops.aten.log10,
args,
kwargs,
rtol=0.001,
atol=0.01,
)

@unittest.skip
def test_aten_log10_2(self):
args = (torch.randint(0, 10, (10, 10)).to(torch.int32),)
kwargs = dict()
Expand All @@ -2386,17 +2390,22 @@ def test_aten_log1p_2(self):
kwargs = dict()
run_export_and_compare(self, torch.ops.aten.log1p, args, kwargs)

@unittest.skip
def test_aten_log2_0(self):
args = (torch.randn((10, 10)).to(torch.float32),)
kwargs = dict()
run_export_and_compare(self, torch.ops.aten.log2, args, kwargs)

@unittest.skip
def test_aten_log2_1(self):
args = (torch.randn((10, 10)).to(torch.float16),)
kwargs = dict()
run_export_and_compare(self, torch.ops.aten.log2, args, kwargs)
run_export_and_compare(
self,
torch.ops.aten.log2,
args,
kwargs,
rtol=0.001,
atol=0.01,
)

def test_aten_log2_2(self):
args = (torch.randint(0, 10, (10, 10)).to(torch.int32),)
Expand Down

0 comments on commit 73b7d7d

Please sign in to comment.