fork-3006-hit2sjtu/dev - Branch - CodSpeedHQ/kornia

Fix type mis-match during mixed-precision training.

Comparing

fork-3006-hit2sjtu/dev

(

3a5fb3f

) with

main

(

d645a84

)

test_aug_2d_thin_plate_spline[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_geometric_test.py

+1%

7.8 s7.8 s

test_aug_2d_elastic_transform[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_geometric_test.py

8.5 s8.5 s

test_aug_2d_solarize[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_intensity_test.py

3 s3 s

test_aug_2d_linear_corner_illumination[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_intensity_test.py

-1%

4.6 s4.6 s

Click on a commit to change the comparison range

Base

main

d645a84

-0.14%

Fix type mis-match during mixed-precision training. Out and patches can be (Float Half), thus cause issue. Signed-off-by: Qiang Wang <wq@sightp.com>

3a5fb3f

1 year ago

by hit2sjtu