Tests

[BUG] Fix for torch==2.6.0. Attempt 2 (#755) #1394

Job	Run time
build-docs	50m 8s
list-test-dirs	4s
build-wheel	48s
run-test (tests/script)	4m 16s
run-test (tests/frontends)	7m 43s
run-test (tests/unit_tests)	19m 22s
run-test (tests/transforms)	3m 23s
run-test (tests/cute_kernel)	13m 22s
run-test (tests/multiprocessing)	3m 56s
run-test (tests/models)	22m 2s
run-test (tests/cute_ops)	8m 19s
run-test (tests/runtime)	4m 41s
run-test (tests/minimal)	3m 35s
run-test (tests/operators)	1h 20m 7s
run-test (tests/utils)	3m 43s
run-test (tests/distributed)	6m 8s
run-test (tests/ir)	8m 35s
run-test (tests/cuda)	9m 36s
run-test (tests/cute_fusion)	45m 48s
run-test (tests/apps)	14m 16s
Pass All Functional Tests	1s
	5h 9m 53s

Provide feedback