ocnn 2.2.7__py3-none-any.whl → 2.3.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (21) hide show
  1. ocnn/__init__.py +1 -1
  2. ocnn/models/resnet.py +2 -2
  3. ocnn/nn/__init__.py +2 -1
  4. ocnn/nn/kernels/__init__.py +14 -0
  5. ocnn/nn/kernels/autotuner.py +416 -0
  6. ocnn/nn/kernels/config.py +67 -0
  7. ocnn/nn/kernels/conv_bwd_implicit_gemm.py +229 -0
  8. ocnn/nn/kernels/conv_bwd_implicit_gemm_splitk.py +347 -0
  9. ocnn/nn/kernels/conv_fwd_implicit_gemm.py +109 -0
  10. ocnn/nn/kernels/conv_fwd_implicit_gemm_splitk.py +150 -0
  11. ocnn/nn/kernels/utils.py +44 -0
  12. ocnn/nn/octree_conv.py +2 -1
  13. ocnn/nn/octree_conv_t.py +148 -0
  14. ocnn/nn/octree_pad.py +4 -4
  15. ocnn/octree/octree.py +218 -109
  16. ocnn/octree/points.py +95 -34
  17. {ocnn-2.2.7.dist-info → ocnn-2.3.0.dist-info}/METADATA +11 -6
  18. {ocnn-2.2.7.dist-info → ocnn-2.3.0.dist-info}/RECORD +21 -12
  19. {ocnn-2.2.7.dist-info → ocnn-2.3.0.dist-info}/WHEEL +1 -1
  20. {ocnn-2.2.7.dist-info → ocnn-2.3.0.dist-info}/licenses/LICENSE +0 -0
  21. {ocnn-2.2.7.dist-info → ocnn-2.3.0.dist-info}/top_level.txt +0 -0
{ocnn-2.2.7.dist-info → ocnn-2.3.0.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- ocnn/__init__.py,sha256=PQFwoI2dNlpDQaMM9jub-VjxFEbutEj2ThIZIHBe9g4,558
1
+ ocnn/__init__.py,sha256=OklIQQwozTf2xldsceBowRnljxu7gQO5VS6mio5CCJQ,558
2
2
  ocnn/dataset.py,sha256=2pRZZdS9l7rRy51CpBInILg-nt6egghUVwgJSxUV0NM,5106
3
3
  ocnn/utils.py,sha256=YmGdO9xfebh9CllOjdkCQMdBo4bOO811InB9G2G0QS0,6521
4
4
  ocnn/models/__init__.py,sha256=KmKgFoWJAoEjcidLeaaJrEvOrxmDfezCcOXFEySICFI,695
@@ -7,30 +7,39 @@ ocnn/models/hrnet.py,sha256=i7wQA5VagBO6lRMM76LB99d6tvzIDgbhn--T5qXH6Zg,6492
7
7
  ocnn/models/image2shape.py,sha256=74PEAGKb0JLRodkADjaEP-Q3s7zOJWQwczf7IpDR61w,4461
8
8
  ocnn/models/lenet.py,sha256=3lDfbFOKNJOKwHRiYMLS_59Pzg5kVpoekFaUtq-FRw8,1708
9
9
  ocnn/models/ounet.py,sha256=su88CyhW6qVp3niGyGi8blBbFkaspk5aN4Xt5BoP8mY,3212
10
- ocnn/models/resnet.py,sha256=CEd_0t6sPnrSkHLxI1xt4S0Dz15RdkGEiqtFXIxnt9Y,1976
10
+ ocnn/models/resnet.py,sha256=mWpLASF4BuIU6tLrpabQizRQEUSSC7vSBLsHspbeJO0,2002
11
11
  ocnn/models/segnet.py,sha256=aUQCJ8g-ygJRs-W7B8ApwNU7Br8wmsDc3UI-RRkJoeY,2503
12
12
  ocnn/models/unet.py,sha256=zQi-e_3iblMq-1FwstRqDTdeG_nIAtGAc6xwzk6RzlA,4018
13
13
  ocnn/modules/__init__.py,sha256=gHFpzESQlofZ0q1ugSoeIajisFdqSs1M5xzfVDpLbmM,1106
14
14
  ocnn/modules/modules.py,sha256=vjsvJpnUyaU1UtZtNuOA9kg0acmZu8C8AKAz0-ytxcA,9543
15
15
  ocnn/modules/resblocks.py,sha256=6siJrsiJaVBFFjGLB3z9zV7Zng2NcMGDeG5RUHMn8dU,5708
16
- ocnn/nn/__init__.py,sha256=7ahdIR8RcQFYPQhxe1EtKwVq0n9xBBhuEAO3n8Zewyc,1780
16
+ ocnn/nn/__init__.py,sha256=rFBRkvFUdAYc2vIV0I86IT4HyBahrvsHJyf-RCs6Vno,1919
17
17
  ocnn/nn/octree2col.py,sha256=GAnUmCm0vqpvdoVhiteDizsjCUNIO0eIFEfWnFu4Z7A,2084
18
18
  ocnn/nn/octree2vox.py,sha256=I0VQpnqJlHaeEbsRBwKb8gnYRana8rSM3l-ypEuKAXI,1474
19
19
  ocnn/nn/octree_align.py,sha256=_6Hr9-ihzZhT70wr15XMc4VYb7Ot2USk0B0mOn3YAlE,1650
20
- ocnn/nn/octree_conv.py,sha256=NnZJrT7wMRUCxeBSIHa54S8sH9umlPNdd3oz0kEs_0k,14513
20
+ ocnn/nn/octree_conv.py,sha256=Ew4fkz1KbholBmBHAHcw-rxw8QoMiMGWDhDV7Aiklr0,14594
21
+ ocnn/nn/octree_conv_t.py,sha256=yhsrSJoUKDjyLa2ZY0-uf9dZ70hdYeYfn5eY96R4BVc,5604
21
22
  ocnn/nn/octree_drop.py,sha256=KzlLNklPFQ9IAnWuvk4F6YQ8jg1P2vwgnsIeUE7eLzc,1908
22
23
  ocnn/nn/octree_dwconv.py,sha256=8sIusaWJMxd3bwdvVVzDcftC2b7jVHkZod3IxgbC0KQ,7128
23
24
  ocnn/nn/octree_gconv.py,sha256=qSAj_TW6vvdc-UKtA7PYSuBf52ijq0CBqLFdVYVpxaE,2849
24
25
  ocnn/nn/octree_interp.py,sha256=wQjh11HeHMTDdzb5csek1JzOtqIM0HLvEj_U8NVC9Ps,6912
25
26
  ocnn/nn/octree_norm.py,sha256=xyUsaTmRDGv-Qxrl3xOQVlN4HA_1wTtTeB4M_bqc-l8,4398
26
- ocnn/nn/octree_pad.py,sha256=T9hqgNJNP8kJJV28puonjahncAl2WrQsRipnLC7TuD4,1284
27
+ ocnn/nn/octree_pad.py,sha256=3zi7lfQkKekUw_Yi3wnFgfDIssVooaJaLFHl1oTLgPU,1282
27
28
  ocnn/nn/octree_pool.py,sha256=smVjdRZkV-GT4qPRjw9cEotbZIq5kceltWMqby0_uvU,6464
29
+ ocnn/nn/kernels/__init__.py,sha256=ChciqrgnHca1twUWn02QYjJOWQ1cIyEEksL7wy2mcGc,479
30
+ ocnn/nn/kernels/autotuner.py,sha256=KR4ZiSZX9l-zL0BW-grlc7MJB5fV92rVejsiNVOMDIY,16105
31
+ ocnn/nn/kernels/config.py,sha256=l1Rw6yyrAv0bUEq-oFpcG7nWIeg7xUqOpmAE-mBAW4I,5253
32
+ ocnn/nn/kernels/conv_bwd_implicit_gemm.py,sha256=8Vb3AeVsm9sMOm-MD0pNlrCUTkDH2hz4AMfWgJTP2Ak,10015
33
+ ocnn/nn/kernels/conv_bwd_implicit_gemm_splitk.py,sha256=9wUOtRH56QHaa9HrFb8aktq7ql8ootdQiZXgHaV3QRQ,14602
34
+ ocnn/nn/kernels/conv_fwd_implicit_gemm.py,sha256=Z3VoPtSGrhrtLQHVVqQ0pzy7d40QQqmCA8skFBi4J5Q,4362
35
+ ocnn/nn/kernels/conv_fwd_implicit_gemm_splitk.py,sha256=kjbEZVFxwmxpJy7Loo_hD9pyAiy2cc-cHIqdRcNjYS0,6250
36
+ ocnn/nn/kernels/utils.py,sha256=2ikyv00Om5mHIupLcVANdNL4CvcSNRxbAQ0AGctC9B8,1216
28
37
  ocnn/octree/__init__.py,sha256=mrO_05mx5-oaSJuPPqwvUyDJeDFFo3iQNana2-rENGc,608
29
- ocnn/octree/octree.py,sha256=ZVr1fbAIB_YhGqwQrZpKu8u-Ft7eYnTee7njgq53GbI,24050
30
- ocnn/octree/points.py,sha256=fkQJmB5hcNFt7NOWDxDEFlg1l76WDBlGHRfG-VQ7_9k,11073
38
+ ocnn/octree/octree.py,sha256=px3fif8XAlqKFtyxjeGSjmXHwML-hWO-UGDHeI6L75A,27994
39
+ ocnn/octree/points.py,sha256=zGrI55l6KOqXGy-1ak6KcUPtlVMDxwK0H4CjOnVdFhc,13148
31
40
  ocnn/octree/shuffled_key.py,sha256=AhXnj3JL1UHxXWrH1QLPJpv_7aKlBI5Eh6CgiJnXE4c,3821
32
- ocnn-2.2.7.dist-info/licenses/LICENSE,sha256=gOZUTXnEc1Od1G6uwCP9_8Lg24YXkCI6iAsMJoGkUlI,1093
33
- ocnn-2.2.7.dist-info/METADATA,sha256=Fx2vIcFGNHaAMpm_8KMVO_bqMwVVZwma8Xui9UJSJkA,5070
34
- ocnn-2.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
35
- ocnn-2.2.7.dist-info/top_level.txt,sha256=ayZdVOnxlOke3kgzAlrRh2IEL_qOudwOaEU3xhjtpZ0,5
36
- ocnn-2.2.7.dist-info/RECORD,,
41
+ ocnn-2.3.0.dist-info/licenses/LICENSE,sha256=gOZUTXnEc1Od1G6uwCP9_8Lg24YXkCI6iAsMJoGkUlI,1093
42
+ ocnn-2.3.0.dist-info/METADATA,sha256=sSgEhB3_hM2qa5fVMCmUZoM2CBU0g4e5Nv6dBx-6UoQ,5270
43
+ ocnn-2.3.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
44
+ ocnn-2.3.0.dist-info/top_level.txt,sha256=ayZdVOnxlOke3kgzAlrRh2IEL_qOudwOaEU3xhjtpZ0,5
45
+ ocnn-2.3.0.dist-info/RECORD,,
{ocnn-2.2.7.dist-info → ocnn-2.3.0.dist-info}/WHEEL RENAMED
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.9.0)
2
+ Generator: setuptools (80.10.2)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5