NotificationsYou must be signed in to change notification settings
Fork26.3k
Star96k

Commitc6a1869

committed

CORRECTION: Actual fixed issue numbers

Fixes#168786Fixes#168787Fixes#168788Fixes#168789Fixes#168790Fixes#168792Fixes#168793Fixes#168794Fixes#168795Fixes#168796Fixes#168797

1 parent0ba333f commitc6a1869Copy full SHA for c6a1869

File tree

1 file changed

-18

lines changed

test
- test_nestedtensor.py

1 file changed

-18

lines changed

`‎test/test_nestedtensor.py‎`

Lines changed: 1 addition & 18 deletions

Original file line number	Diff line number	Diff line change
`@@ -6724,12 +6724,7 @@ def fn(values, same_size):`
`6724`	`6724`	`check_results(fn,compiled_fn,generate_inp(20))`
`6725`	`6725`	`self.assertEqual(compile_counter.frame_count,frame_count_2)`
`6726`	`6726`
`6727`		`-# Note 1: Math fallback doesn't work with bfloat16 on CUDA`
`6728`		`-# Note 2: ROCm doesn't support flash attention or mem_efficient attention for NT`
`6729`		`-@unittest.skipIf(`
`6730`		`-TEST_WITH_ROCM,`
`6731`		`-"ROCm doesn't support flash attention or mem_efficient attention for NT",`
`6732`		`- )`
	`6727`	`+# Note: Math fallback doesn't work with bfloat16 on CUDA`
`6733`	`6728`	`@tf32_on_and_off(0.005)`
`6734`	`6729`	`@dtypes(`
`6735`	`6730`	`*(`
`@@ -6999,9 +6994,7 @@ def check_forward_backward(skip_backward=False):`
`6999`	`6994`
`7000`	`6995`	`@skipIfTorchDynamo("SDPA test compiles internally")`
`7001`	`6996`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7002`		`-# Guarding with sqrt() doesn't work on ROCm?`
`7003`	`6997`	`@xfailIfWindows`
`7004`		`-@skipCUDAIfRocm`
`7005`	`6998`	`@onlyCUDA`
`7006`	`6999`	`@dtypes(`
`7007`	`7000`	`*(`
`@@ -7188,8 +7181,6 @@ def in_proj(input_packed, qkv_linear=qkv_linear):`
`7188`	`7181`	`@decorateIf(xfailIfWindows,lambdaparams:params["dtype"]==torch.float32)`
`7189`	`7182`	`@skipIfTorchDynamo("SDPA test compiles internally")`
`7190`	`7183`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7191`		`-# mha_varlen_fwd not supported on ROCm`
`7192`		`-@skipCUDAIfRocm`
`7193`	`7184`	`@onlyCUDA`
`7194`	`7185`	`@dtypes(`
`7195`	`7186`	`*(`
`@@ -7220,7 +7211,6 @@ def f(values, offsets):`
`7220`	`7211`	`"Platform doesn't support flash or mem-efficient attention",`
`7221`	`7212`	`)`
`7222`	`7213`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7223`		`-@skipCUDAIfRocm`
`7224`	`7214`	`@onlyCUDA`
`7225`	`7215`	`@skipIfTorchDynamo()`
`7226`	`7216`	`deftest_sdpa_autocast(self,device):`
`@@ -7303,7 +7293,6 @@ def get_values():`
`7303`	`7293`	`"Platform doesn't support flash or mem-efficient attention",`
`7304`	`7294`	`)`
`7305`	`7295`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7306`		`-@skipCUDAIfRocm`
`7307`	`7296`	`@onlyCUDA`
`7308`	`7297`	`@skipIfTorchDynamo()`
`7309`	`7298`	`deftest_sdpa_flop_counter(self,device):`
`@@ -7726,7 +7715,6 @@ def test_jagged_padded_dense_conversion_kernels(self, device, dtype):`
`7726`	`7715`	`@dtypes(torch.float32)`
`7727`	`7716`	`@skipIfTorchDynamo("Test compiles internally")`
`7728`	`7717`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7729`		`-@skipCUDAIfRocm`
`7730`	`7718`	`deftest_compile_preserves_metadata_cache(self,device,dtype):`
`7731`	`7719`	`# shape (B, *, D)`
`7732`	`7720`	`nt=random_nt_from_dims(`
`@@ -7753,7 +7741,6 @@ def f(nt):`
`7753`	`7741`	`@dtypes(torch.float32)`
`7754`	`7742`	`@skipIfTorchDynamo("Test compiles internally")`
`7755`	`7743`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7756`		`-@skipCUDAIfRocm`
`7757`	`7744`	`deftest_compile_with_dynamic_max_seq_len(self,device,dtype):`
`7758`	`7745`	`# shape (B, *, D)`
`7759`	`7746`	`# max seq len: 18`
`@@ -7786,7 +7773,6 @@ def f(nt):`
`7786`	`7773`	`@dtypes(torch.float32)`
`7787`	`7774`	`@skipIfTorchDynamo("Test compiles internally")`
`7788`	`7775`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7789`		`-@skipCUDAIfRocm`
`7790`	`7776`	`deftest_compile_with_dynamic_min_seq_len(self,device,dtype):`
`7791`	`7777`	`# shape (B, *, D)`
`7792`	`7778`	`# min seq len: 7`
`@@ -7819,7 +7805,6 @@ def f(nt):`
`7819`	`7805`	`@dtypes(torch.float32)`
`7820`	`7806`	`@skipIfTorchDynamo("Test compiles internally")`
`7821`	`7807`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7822`		`-@skipCUDAIfRocm`
`7823`	`7808`	`deftest_compile_with_propagated_dynamic_max_seq_len(self,device,dtype):`
`7824`	`7809`	`# shape (B, *, D)`
`7825`	`7810`	`# max seq len: 18`
`@@ -7946,7 +7931,6 @@ def test_to_padded_tensor(self, device, dtype, nt_dim, requires_grad):`
`7946`	`7931`	`@torch._dynamo.utils.disable_cache_limit()`
`7947`	`7932`	`@skipIfTorchDynamo("SDPA test compiles internally")`
`7948`	`7933`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`7949`		`-@skipCUDAIfRocm`
`7950`	`7934`	`@dtypes(torch.float32,torch.double,torch.half)`
`7951`	`7935`	`@parametrize("nt_dim", [2,3,4])`
`7952`	`7936`	`@parametrize("requires_grad", [False,True])`
`@@ -8048,7 +8032,6 @@ def _g(nt):`
`8048`	`8032`	`@dtypes(torch.float32)`
`8049`	`8033`	`@skipIfTorchDynamo("Test compiles internally")`
`8050`	`8034`	`@skipCUDAIf(notSM70OrLater,"GPU capability is < SM70")`
`8051`		`-@skipCUDAIfRocm`
`8052`	`8035`	`deftest_compile_padded_dense_conversion_preserves_metadata_cache(`
`8053`	`8036`	`self,device,dtype`
`8054`	`8037`	`):`

0 commit comments

Comments

(0)

Movatterモバイル変換

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commitc6a1869

File tree

1 file changed

1 file changed

`‎test/test_nestedtensor.py‎`

0 commit comments