brishen commited on
Commit
d99fb9f
·
verified ·
1 Parent(s): adcf47f

Upload the GAIC-VGG16 NPU2 bundle for taconite-gaic 0.1.0

Browse files
.gitattributes CHANGED
@@ -33,3 +33,12 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ kernels/fc1.xclbin filter=lfs diff=lfs merge=lfs -text
37
+ kernels/m1024_k1536_n384_lda1152.xclbin filter=lfs diff=lfs merge=lfs -text
38
+ kernels/m1024_k1536_n768_lda1152.xclbin filter=lfs diff=lfs merge=lfs -text
39
+ kernels/m1024_k2304_n256_lda768.xclbin filter=lfs diff=lfs merge=lfs -text
40
+ kernels/m1024_k2304_n512_lda1536.xclbin filter=lfs diff=lfs merge=lfs -text
41
+ kernels/m2048_k256_n512_lda0.xclbin filter=lfs diff=lfs merge=lfs -text
42
+ kernels/m512_k1536_n512_lda768.xclbin filter=lfs diff=lfs merge=lfs -text
43
+ kernels/m512_k2304_n512_lda768.xclbin filter=lfs diff=lfs merge=lfs -text
44
+ kernels/m512_k4608_n512_lda1536.xclbin filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - iron
5
+ - amd-npu
6
+ - ryzen-ai
7
+ - xdna
8
+ - image-cropping
9
+ - npu2
10
+ ---
11
+
12
+ # GAIC-VGG16 crop scoring on the AMD NPU
13
+
14
+ An [IRON](https://github.com/amd/IRON) export of [https://github.com/bo-zhang-cs/GAIC-Pytorch](https://github.com/bo-zhang-cs/GAIC-Pytorch) for AMD Ryzen AI NPUs: the compiled NPU kernels (`.xclbin` + instruction streams) and the packed weights the [`taconite-gaic`](https://crates.io/crates/taconite-gaic) Rust runtime replays, over XRT or directly over the `amdxdna` driver.
15
+
16
+ The kernels are compiled for **NPU2 (AIE2P: Strix Point, Strix Halo, Krackan)** and will not load on NPU1 (Phoenix, Hawk Point).
17
+
18
+ ## Download
19
+
20
+ ```bash
21
+ hf download brishen/iron-gaic-vgg16-npu2 --local-dir gaic
22
+ # or, from an IRON checkout:
23
+ python scripts/hf_models.py download gaic --repo brishen/iron-gaic-vgg16-npu2 --out gaic
24
+ ```
25
+
26
+ ## Usage
27
+
28
+ ```bash
29
+ cargo install taconite-gaic # over XRT
30
+ # or, with no XRT at all (straight to the amdxdna driver):
31
+ cargo install taconite-gaic --no-default-features --features cli,direct
32
+ gaic gaic check # against the bundle's reference image
33
+ gaic gaic crop --out crops/ photo.jpg # best crop overall and at 1:1, 4:3, 16:9
34
+ ```
35
+
36
+ See [`taconite-gaic`](https://crates.io/crates/taconite-gaic) ([docs](https://docs.rs/taconite-gaic), [source](https://github.com/Brishen/taconite)) for the full API. The bundle is exported by IRON's `iron/applications/gaic`.
37
+
38
+ ## Provenance
39
+
40
+ - Upstream model: [https://github.com/bo-zhang-cs/GAIC-Pytorch](https://github.com/bo-zhang-cs/GAIC-Pytorch) (license: `mit`; its terms apply to these weights)
41
+ - IRON commit: `be4bb37`
42
+ - Uploaded: 2026-09-23
43
+ - Files: 21, 29.2 MB
kernels/fc1.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78ccfeec316f03cb3d7bed482276515eff7dcebfb1691690c03e3f72f4f35250
3
+ size 7456
kernels/fc1.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3641e68e9c4bf682a384e6ceaaac82f748231588ce2c84bb877c9177af3c04dd
3
+ size 202303
kernels/m1024_k1536_n384_lda1152.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04aa591ffd8d4ca9b7f9e4d2043a58a81a369e80d0833d1a397786cad8bfac85
3
+ size 4504
kernels/m1024_k1536_n384_lda1152.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4629b2ef02f8e99434af3772a3358bcc3e78bfe3b50dd85cf5ce55f7242c403
3
+ size 143759
kernels/m1024_k1536_n768_lda1152.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b03582770a9c480f870e8139b9890cdda57f52d83c44f79f4ecf0ae5c7700a8
3
+ size 7456
kernels/m1024_k1536_n768_lda1152.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81811bba1dd7fff9577615d8383160e26e8ac02510f2b4c966e48eba73074083
3
+ size 189503
kernels/m1024_k2304_n256_lda768.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:725aa5e57093ec33effe57402462ef4a74a25f4d2434d499215e313cf88ee6b9
3
+ size 3168
kernels/m1024_k2304_n256_lda768.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fbdfd0a5a44f836ac20ac74a870188191fb356d6baba57e6d4290cba0219bdc
3
+ size 146319
kernels/m1024_k2304_n512_lda1536.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:769a0bfefbad4fde22c3a651fa0d4eb95d2ebd439c6e0c8815c4c6914e87c78c
3
+ size 5840
kernels/m1024_k2304_n512_lda1536.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73c841778255382bdb75505bc422f8131e21adc0072ea578a7a63a5f2561584a
3
+ size 286271
kernels/m2048_k256_n512_lda0.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:915a6acda7d9b6a8d67ba7ef0f66e588dac1049300e1cddf5ef08507dc1e872c
3
+ size 3168
kernels/m2048_k256_n512_lda0.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fded181aeff009d6bed97bdf6cbf95d705a2fdecc99816ac9d003a52fb0fe4f1
3
+ size 104335
kernels/m512_k1536_n512_lda768.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22f31fd260befc05d3a30f25e3e1423a7bc7168c387f01ebd6082dfb1c07137e
3
+ size 5840
kernels/m512_k1536_n512_lda768.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b65818a5fd025261e7aa0660b0dde52f75094d38e2aeded08b4b2aeaf85f50b6
3
+ size 186431
kernels/m512_k2304_n512_lda768.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05a1c7f67c23c38689f76b1d7b718b083c6d3ee2f13157184e1000e765b12f58
3
+ size 5840
kernels/m512_k2304_n512_lda768.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e9502295c9653ad891034840a67cb492984b9c12b2be8f8e64959afb59a7a74
3
+ size 237119
kernels/m512_k4608_n512_lda1536.insts.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d68f1705d0121737018e0e74adc33bc47a0dde84645532eff7d06c5892ec4e10
3
+ size 5840
kernels/m512_k4608_n512_lda1536.xclbin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a46a920ddc5917c988b8e0784b450fcb70d06dae7d374f31509102af1786c16
3
+ size 205631
manifest.txt ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ version 1
2
+ xclbin m2048_k256_n512_lda0 kernels/m2048_k256_n512_lda0.xclbin MLIR_AIE
3
+ kernel m2048_k256_n512_lda0 ctx=m2048_k256_n512_lda0 insts=kernels/m2048_k256_n512_lda0.insts.bin M=2048 K=256 N=512 a_elems=524288 b_bytes=147456 c_elems=1048576
4
+ xclbin m1024_k1536_n384_lda1152 kernels/m1024_k1536_n384_lda1152.xclbin MLIR_AIE
5
+ kernel m1024_k1536_n384_lda1152 ctx=m1024_k1536_n384_lda1152 insts=kernels/m1024_k1536_n384_lda1152.insts.bin M=1024 K=1536 N=384 a_elems=1180032 b_bytes=663552 c_elems=393216
6
+ xclbin m1024_k1536_n768_lda1152 kernels/m1024_k1536_n768_lda1152.xclbin MLIR_AIE
7
+ kernel m1024_k1536_n768_lda1152 ctx=m1024_k1536_n768_lda1152 insts=kernels/m1024_k1536_n768_lda1152.insts.bin M=1024 K=1536 N=768 a_elems=1180032 b_bytes=1327104 c_elems=786432
8
+ xclbin m1024_k2304_n512_lda1536 kernels/m1024_k2304_n512_lda1536.xclbin MLIR_AIE
9
+ kernel m1024_k2304_n512_lda1536 ctx=m1024_k2304_n512_lda1536 insts=kernels/m1024_k2304_n512_lda1536.insts.bin M=1024 K=2304 N=512 a_elems=1573632 b_bytes=1327104 c_elems=524288
10
+ xclbin m512_k1536_n512_lda768 kernels/m512_k1536_n512_lda768.xclbin MLIR_AIE
11
+ kernel m512_k1536_n512_lda768 ctx=m512_k1536_n512_lda768 insts=kernels/m512_k1536_n512_lda768.insts.bin M=512 K=1536 N=512 a_elems=393984 b_bytes=884736 c_elems=262144
12
+ xclbin m1024_k2304_n256_lda768 kernels/m1024_k2304_n256_lda768.xclbin MLIR_AIE
13
+ kernel m1024_k2304_n256_lda768 ctx=m1024_k2304_n256_lda768 insts=kernels/m1024_k2304_n256_lda768.insts.bin M=1024 K=2304 N=256 a_elems=787968 b_bytes=663552 c_elems=262144
14
+ xclbin m512_k2304_n512_lda768 kernels/m512_k2304_n512_lda768.xclbin MLIR_AIE
15
+ kernel m512_k2304_n512_lda768 ctx=m512_k2304_n512_lda768 insts=kernels/m512_k2304_n512_lda768.insts.bin M=512 K=2304 N=512 a_elems=394752 b_bytes=1327104 c_elems=262144
16
+ xclbin m512_k4608_n512_lda1536 kernels/m512_k4608_n512_lda1536.xclbin MLIR_AIE
17
+ kernel m512_k4608_n512_lda1536 ctx=m512_k4608_n512_lda1536 insts=kernels/m512_k4608_n512_lda1536.insts.bin M=512 K=4608 N=512 a_elems=789504 b_bytes=2654208 c_elems=262144
18
+ xclbin fc1 kernels/fc1.xclbin MLIR_AIE
19
+ kernel fc1 ctx=fc1 insts=kernels/fc1.insts.bin M=256 K=5376 N=768 a_elems=1376256 b_bytes=4644864 c_elems=196608
20
+ conv conv1_1 kernel=m2048_k256_n512_lda0 C=3 OC=64 pool_before=0 m_chunk=2048 P=8 window=1 D=9 b=conv.conv1_1.b bias=conv.conv1_1.bias
21
+ conv conv1_2 kernel=m1024_k1536_n384_lda1152 C=64 OC=64 pool_before=0 m_chunk=1024 P=6 window=0 D=192 b=conv.conv1_2.b bias=conv.conv1_2.bias
22
+ conv conv2_1 kernel=m1024_k1536_n768_lda1152 C=64 OC=128 pool_before=1 m_chunk=1024 P=6 window=0 D=192 b=conv.conv2_1.b bias=conv.conv2_1.bias
23
+ conv conv2_2 kernel=m1024_k2304_n512_lda1536 C=128 OC=128 pool_before=0 m_chunk=1024 P=4 window=0 D=384 b=conv.conv2_2.b bias=conv.conv2_2.bias
24
+ conv conv3_1 kernel=m512_k1536_n512_lda768 C=128 OC=256 pool_before=1 m_chunk=512 P=2 window=0 D=384 b=conv.conv3_1.b bias=conv.conv3_1.bias
25
+ conv conv3_2 kernel=m1024_k2304_n256_lda768 C=256 OC=256 pool_before=0 m_chunk=1024 P=1 window=0 D=768 b=conv.conv3_2.b bias=conv.conv3_2.bias
26
+ conv conv3_3 kernel=m1024_k2304_n256_lda768 C=256 OC=256 pool_before=0 m_chunk=1024 P=1 window=0 D=768 b=conv.conv3_3.b bias=conv.conv3_3.bias
27
+ conv conv4_1 kernel=m512_k2304_n512_lda768 C=256 OC=512 pool_before=1 m_chunk=512 P=1 window=0 D=768 b=conv.conv4_1.b bias=conv.conv4_1.bias
28
+ conv conv4_2 kernel=m512_k4608_n512_lda1536 C=512 OC=512 pool_before=0 m_chunk=512 P=1 window=0 D=1536 b=conv.conv4_2.b bias=conv.conv4_2.bias
29
+ conv conv4_3 kernel=m512_k4608_n512_lda1536 C=512 OC=512 pool_before=0 m_chunk=512 P=1 window=0 D=1536 b=conv.conv4_3.b bias=conv.conv4_3.bias
30
+ conv conv5_1 kernel=m512_k4608_n512_lda1536 C=512 OC=512 pool_before=1 m_chunk=512 P=1 window=0 D=1536 b=conv.conv5_1.b bias=conv.conv5_1.bias
31
+ conv conv5_2 kernel=m512_k4608_n512_lda1536 C=512 OC=512 pool_before=0 m_chunk=512 P=1 window=0 D=1536 b=conv.conv5_2.b bias=conv.conv5_2.bias
32
+ conv conv5_3 kernel=m512_k4608_n512_lda1536 C=512 OC=512 pool_before=0 m_chunk=512 P=1 window=0 D=1536 b=conv.conv5_3.b bias=conv.conv5_3.bias
33
+ param f3 conv4_3
34
+ param f4 conv5_3
35
+ dimred w=dimred.w b=dimred.b out=32 in=1536
36
+ fc1 kernel=fc1 b=fc1.b bias=fc1.bias k=5184 k_pad=5376 n=768 m=256
37
+ fc2 w=fc2.w b=fc2.b out=128 in=768
38
+ fc3 w=fc3.w b=fc3.b in=128
39
+ param align_size 9
40
+ param spatial_scale 0.0625
41
+ ref image=211958.jpg w=384 h=256 anchors=83 src_w=1024 src_h=683
tensors.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5bdb2016e3f9b3a7cbfe1479d7f1b3c4a5183c3c71959f68b23099a8fb942f9f
3
+ size 28909260
tensors.txt ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ conv.conv1_1.b u8 147456 0 147456
2
+ conv.conv1_1.bias f32 64 147456 256
3
+ conv.conv1_2.b u8 663552 147712 663552
4
+ conv.conv1_2.bias f32 64 811264 256
5
+ conv.conv2_1.b u8 1327104 811520 1327104
6
+ conv.conv2_1.bias f32 128 2138624 512
7
+ conv.conv2_2.b u8 1327104 2139136 1327104
8
+ conv.conv2_2.bias f32 128 3466240 512
9
+ conv.conv3_1.b u8 884736 3466752 884736
10
+ conv.conv3_1.bias f32 256 4351488 1024
11
+ conv.conv3_2.b u8 663552 4352512 663552
12
+ conv.conv3_2.bias f32 256 5016064 1024
13
+ conv.conv3_3.b u8 663552 5017088 663552
14
+ conv.conv3_3.bias f32 256 5680640 1024
15
+ conv.conv4_1.b u8 1327104 5681664 1327104
16
+ conv.conv4_1.bias f32 512 7008768 2048
17
+ conv.conv4_2.b u8 2654208 7010816 2654208
18
+ conv.conv4_2.bias f32 512 9665024 2048
19
+ conv.conv4_3.b u8 2654208 9667072 2654208
20
+ conv.conv4_3.bias f32 512 12321280 2048
21
+ conv.conv5_1.b u8 2654208 12323328 2654208
22
+ conv.conv5_1.bias f32 512 14977536 2048
23
+ conv.conv5_2.b u8 2654208 14979584 2654208
24
+ conv.conv5_2.bias f32 512 17633792 2048
25
+ conv.conv5_3.b u8 2654208 17635840 2654208
26
+ conv.conv5_3.bias f32 512 20290048 2048
27
+ dimred.w f32 32,1536 20292096 196608
28
+ dimred.b f32 32 20488704 128
29
+ fc1.b u8 4644864 20488832 4644864
30
+ fc1.bias f32 768 25133696 3072
31
+ fc2.w f32 128,768 25136768 393216
32
+ fc2.b f32 128 25529984 512
33
+ fc3.w f32 128 25530496 512
34
+ fc3.b f32 1 25531008 4
35
+ ref.src_rgb u8 683,1024,3 25531072 2098176
36
+ ref.input_chw f32 3,256,384 27629248 1179648
37
+ ref.red_cpu f32 32,16,24 28808896 49152
38
+ ref.red_npu f32 32,16,24 28858048 49152
39
+ ref.anchors i32 83,4 28907200 1328
40
+ ref.scores_cpu f32 83 28908544 332
41
+ ref.scores_npu f32 83 28908928 332