Hugging Face's logo

Kernels:
flashrt
/
flashrt-spatiotemporal-layout

Kernel card Files
xet
 Community
liangsu9988 commited on
Commit
728a92e
·
verified ·
1 Parent(s): 9d82137

Uploaded using `kernel-builder`.

Browse files
Files changed (16) hide show
  1. build/torch211-cxx11-cu128-x86_64-linux/__init__.py +19 -0
  2. build/torch211-cxx11-cu128-x86_64-linux/{_flashrt_spatiotemporal_layout_cuda_cf903dd.abi3.so → _flashrt_spatiotemporal_layout_cuda_5de4768.abi3.so} +2 -2
  3. build/torch211-cxx11-cu128-x86_64-linux/_ops.py +3 -3
  4. build/torch211-cxx11-cu128-x86_64-linux/metadata.json +1 -1
  5. build/torch211-cxx11-cu130-x86_64-linux/__init__.py +19 -0
  6. build/torch211-cxx11-cu130-x86_64-linux/{_flashrt_spatiotemporal_layout_cuda_cf903dd.abi3.so → _flashrt_spatiotemporal_layout_cuda_5de4768.abi3.so} +2 -2
  7. build/torch211-cxx11-cu130-x86_64-linux/_ops.py +3 -3
  8. build/torch211-cxx11-cu130-x86_64-linux/metadata.json +3 -2
  9. build/torch212-cxx11-cu130-x86_64-linux/__init__.py +19 -0
  10. build/torch212-cxx11-cu130-x86_64-linux/{_flashrt_spatiotemporal_layout_cuda_cf903dd.abi3.so → _flashrt_spatiotemporal_layout_cuda_5de4768.abi3.so} +2 -2
  11. build/torch212-cxx11-cu130-x86_64-linux/_ops.py +3 -3
  12. build/torch212-cxx11-cu130-x86_64-linux/metadata.json +3 -2
  13. build/torch212-cxx11-cu132-x86_64-linux/__init__.py +19 -0
  14. build/torch212-cxx11-cu132-x86_64-linux/{_flashrt_spatiotemporal_layout_cuda_cf903dd.abi3.so → _flashrt_spatiotemporal_layout_cuda_5de4768.abi3.so} +2 -2
  15. build/torch212-cxx11-cu132-x86_64-linux/_ops.py +3 -3
  16. build/torch212-cxx11-cu132-x86_64-linux/metadata.json +3 -2
build/torch211-cxx11-cu128-x86_64-linux/__init__.py CHANGED
@@ -21,6 +21,15 @@ def _ncdhw_to_blc_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
21
  return None
22
 
23
 
 
 
 
 
 
 
 
 
 
24
  @torch.library.register_fake(add_op_namespace_prefix("time_unshuffle2_bf16"))
25
  def _time_unshuffle2_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
26
  _check_ncdhw(x, "x")
@@ -58,6 +67,15 @@ def ncdhw_to_blc_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch
58
  return out
59
 
60
 
 
 
 
 
 
 
 
 
 
61
  def time_unshuffle2_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
62
  """Convert BF16 (B, 2C, T, H, W) into (B, C, 2T, H, W)."""
63
 
@@ -85,6 +103,7 @@ def update_cache2_ncdhw_bf16(cur: torch.Tensor, prev: torch.Tensor, out: torch.T
85
 
86
  __all__ = [
87
  "ncdhw_to_blc_bf16",
 
88
  "time_unshuffle2_bf16",
89
  "add_bias_ncdhw_bf16",
90
  "update_cache2_ncdhw_bf16",
 
21
  return None
22
 
23
 
24
+ @torch.library.register_fake(add_op_namespace_prefix("patch_im2col_bf16"))
25
+ def _patch_im2col_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
26
+ if x.dim() != 4 or x.shape[1:] != (224, 224, 3):
27
+ raise RuntimeError("x must have shape (num_views, 224, 224, 3)")
28
+ if out.shape != (x.shape[0] * 256, 588):
29
+ raise RuntimeError("out must have shape (num_views * 256, 588)")
30
+ return None
31
+
32
+
33
  @torch.library.register_fake(add_op_namespace_prefix("time_unshuffle2_bf16"))
34
  def _time_unshuffle2_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
35
  _check_ncdhw(x, "x")
 
67
  return out
68
 
69
 
70
+ def patch_im2col_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
71
+ """Convert BF16 NHWC images into flattened 14x14 patch rows."""
72
+
73
+ if out is None:
74
+ out = torch.empty((x.shape[0] * 256, 588), device=x.device, dtype=x.dtype)
75
+ ops.patch_im2col_bf16(x, out)
76
+ return out
77
+
78
+
79
  def time_unshuffle2_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
80
  """Convert BF16 (B, 2C, T, H, W) into (B, C, 2T, H, W)."""
81
 
 
103
 
104
  __all__ = [
105
  "ncdhw_to_blc_bf16",
106
+ "patch_im2col_bf16",
107
  "time_unshuffle2_bf16",
108
  "add_bias_ncdhw_bf16",
109
  "update_cache2_ncdhw_bf16",
build/torch211-cxx11-cu128-x86_64-linux/{_flashrt_spatiotemporal_layout_cuda_cf903dd.abi3.so → _flashrt_spatiotemporal_layout_cuda_5de4768.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:18a84407f77c3f0d010257989c79bfc295e38c0f7eaa2f39d1b96da2b99cde15
3
- size 2610792
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:280efdb5e795d088a9d44e031b64b3d652a70cb070325579c2623690b8953b3d
3
+ size 727528
build/torch211-cxx11-cu128-x86_64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flashrt_spatiotemporal_layout_cuda_cf903dd
3
- ops = torch.ops._flashrt_spatiotemporal_layout_cuda_cf903dd
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flashrt_spatiotemporal_layout_cuda_cf903dd::{op_name}"
 
1
  import torch
2
+ from . import _flashrt_spatiotemporal_layout_cuda_5de4768
3
+ ops = torch.ops._flashrt_spatiotemporal_layout_cuda_5de4768
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flashrt_spatiotemporal_layout_cuda_5de4768::{op_name}"
build/torch211-cxx11-cu128-x86_64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flashrt-spatiotemporal-layout",
3
- "id": "_flashrt_spatiotemporal_layout_cuda_cf903dd",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
1
  {
2
  "name": "flashrt-spatiotemporal-layout",
3
+ "id": "_flashrt_spatiotemporal_layout_cuda_5de4768",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
build/torch211-cxx11-cu130-x86_64-linux/__init__.py CHANGED
@@ -21,6 +21,15 @@ def _ncdhw_to_blc_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
21
  return None
22
 
23
 
 
 
 
 
 
 
 
 
 
24
  @torch.library.register_fake(add_op_namespace_prefix("time_unshuffle2_bf16"))
25
  def _time_unshuffle2_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
26
  _check_ncdhw(x, "x")
@@ -58,6 +67,15 @@ def ncdhw_to_blc_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch
58
  return out
59
 
60
 
 
 
 
 
 
 
 
 
 
61
  def time_unshuffle2_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
62
  """Convert BF16 (B, 2C, T, H, W) into (B, C, 2T, H, W)."""
63
 
@@ -85,6 +103,7 @@ def update_cache2_ncdhw_bf16(cur: torch.Tensor, prev: torch.Tensor, out: torch.T
85
 
86
  __all__ = [
87
  "ncdhw_to_blc_bf16",
 
88
  "time_unshuffle2_bf16",
89
  "add_bias_ncdhw_bf16",
90
  "update_cache2_ncdhw_bf16",
 
21
  return None
22
 
23
 
24
+ @torch.library.register_fake(add_op_namespace_prefix("patch_im2col_bf16"))
25
+ def _patch_im2col_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
26
+ if x.dim() != 4 or x.shape[1:] != (224, 224, 3):
27
+ raise RuntimeError("x must have shape (num_views, 224, 224, 3)")
28
+ if out.shape != (x.shape[0] * 256, 588):
29
+ raise RuntimeError("out must have shape (num_views * 256, 588)")
30
+ return None
31
+
32
+
33
  @torch.library.register_fake(add_op_namespace_prefix("time_unshuffle2_bf16"))
34
  def _time_unshuffle2_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
35
  _check_ncdhw(x, "x")
 
67
  return out
68
 
69
 
70
+ def patch_im2col_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
71
+ """Convert BF16 NHWC images into flattened 14x14 patch rows."""
72
+
73
+ if out is None:
74
+ out = torch.empty((x.shape[0] * 256, 588), device=x.device, dtype=x.dtype)
75
+ ops.patch_im2col_bf16(x, out)
76
+ return out
77
+
78
+
79
  def time_unshuffle2_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
80
  """Convert BF16 (B, 2C, T, H, W) into (B, C, 2T, H, W)."""
81
 
 
103
 
104
  __all__ = [
105
  "ncdhw_to_blc_bf16",
106
+ "patch_im2col_bf16",
107
  "time_unshuffle2_bf16",
108
  "add_bias_ncdhw_bf16",
109
  "update_cache2_ncdhw_bf16",
build/torch211-cxx11-cu130-x86_64-linux/{_flashrt_spatiotemporal_layout_cuda_cf903dd.abi3.so → _flashrt_spatiotemporal_layout_cuda_5de4768.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b1432b4f402b24b0cf9dda42334059b9ea83ed947cc99e8714fb80a9cfba38e5
3
- size 2513352
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b925abfb8b8b8dadffaeba05b23a0059612b6ccc83ddcd75a620bcb606daad6f
3
+ size 696272
build/torch211-cxx11-cu130-x86_64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flashrt_spatiotemporal_layout_cuda_cf903dd
3
- ops = torch.ops._flashrt_spatiotemporal_layout_cuda_cf903dd
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flashrt_spatiotemporal_layout_cuda_cf903dd::{op_name}"
 
1
  import torch
2
+ from . import _flashrt_spatiotemporal_layout_cuda_5de4768
3
+ ops = torch.ops._flashrt_spatiotemporal_layout_cuda_5de4768
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flashrt_spatiotemporal_layout_cuda_5de4768::{op_name}"
build/torch211-cxx11-cu130-x86_64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flashrt-spatiotemporal-layout",
3
- "id": "_flashrt_spatiotemporal_layout_cuda_cf903dd",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
@@ -9,7 +9,8 @@
9
  "archs": [
10
  "10.0",
11
  "11.0",
12
- "12.0+PTX",
 
13
  "7.5",
14
  "8.0",
15
  "8.6",
 
1
  {
2
  "name": "flashrt-spatiotemporal-layout",
3
+ "id": "_flashrt_spatiotemporal_layout_cuda_5de4768",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
9
  "archs": [
10
  "10.0",
11
  "11.0",
12
+ "12.0",
13
+ "12.1+PTX",
14
  "7.5",
15
  "8.0",
16
  "8.6",
build/torch212-cxx11-cu130-x86_64-linux/__init__.py CHANGED
@@ -21,6 +21,15 @@ def _ncdhw_to_blc_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
21
  return None
22
 
23
 
 
 
 
 
 
 
 
 
 
24
  @torch.library.register_fake(add_op_namespace_prefix("time_unshuffle2_bf16"))
25
  def _time_unshuffle2_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
26
  _check_ncdhw(x, "x")
@@ -58,6 +67,15 @@ def ncdhw_to_blc_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch
58
  return out
59
 
60
 
 
 
 
 
 
 
 
 
 
61
  def time_unshuffle2_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
62
  """Convert BF16 (B, 2C, T, H, W) into (B, C, 2T, H, W)."""
63
 
@@ -85,6 +103,7 @@ def update_cache2_ncdhw_bf16(cur: torch.Tensor, prev: torch.Tensor, out: torch.T
85
 
86
  __all__ = [
87
  "ncdhw_to_blc_bf16",
 
88
  "time_unshuffle2_bf16",
89
  "add_bias_ncdhw_bf16",
90
  "update_cache2_ncdhw_bf16",
 
21
  return None
22
 
23
 
24
+ @torch.library.register_fake(add_op_namespace_prefix("patch_im2col_bf16"))
25
+ def _patch_im2col_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
26
+ if x.dim() != 4 or x.shape[1:] != (224, 224, 3):
27
+ raise RuntimeError("x must have shape (num_views, 224, 224, 3)")
28
+ if out.shape != (x.shape[0] * 256, 588):
29
+ raise RuntimeError("out must have shape (num_views * 256, 588)")
30
+ return None
31
+
32
+
33
  @torch.library.register_fake(add_op_namespace_prefix("time_unshuffle2_bf16"))
34
  def _time_unshuffle2_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
35
  _check_ncdhw(x, "x")
 
67
  return out
68
 
69
 
70
+ def patch_im2col_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
71
+ """Convert BF16 NHWC images into flattened 14x14 patch rows."""
72
+
73
+ if out is None:
74
+ out = torch.empty((x.shape[0] * 256, 588), device=x.device, dtype=x.dtype)
75
+ ops.patch_im2col_bf16(x, out)
76
+ return out
77
+
78
+
79
  def time_unshuffle2_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
80
  """Convert BF16 (B, 2C, T, H, W) into (B, C, 2T, H, W)."""
81
 
 
103
 
104
  __all__ = [
105
  "ncdhw_to_blc_bf16",
106
+ "patch_im2col_bf16",
107
  "time_unshuffle2_bf16",
108
  "add_bias_ncdhw_bf16",
109
  "update_cache2_ncdhw_bf16",
build/torch212-cxx11-cu130-x86_64-linux/{_flashrt_spatiotemporal_layout_cuda_cf903dd.abi3.so → _flashrt_spatiotemporal_layout_cuda_5de4768.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3c6dbbf811af69ff253978b96b02d4e9cabec4f20f2ad5a887aca31cf5267d14
3
- size 2518880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c470dfb5ff53452ad0eddb0e5a4204dcdd151c6058e3b7b1ef7b77c765aaa91e
3
+ size 702464
build/torch212-cxx11-cu130-x86_64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flashrt_spatiotemporal_layout_cuda_cf903dd
3
- ops = torch.ops._flashrt_spatiotemporal_layout_cuda_cf903dd
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flashrt_spatiotemporal_layout_cuda_cf903dd::{op_name}"
 
1
  import torch
2
+ from . import _flashrt_spatiotemporal_layout_cuda_5de4768
3
+ ops = torch.ops._flashrt_spatiotemporal_layout_cuda_5de4768
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flashrt_spatiotemporal_layout_cuda_5de4768::{op_name}"
build/torch212-cxx11-cu130-x86_64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flashrt-spatiotemporal-layout",
3
- "id": "_flashrt_spatiotemporal_layout_cuda_cf903dd",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
@@ -9,7 +9,8 @@
9
  "archs": [
10
  "10.0",
11
  "11.0",
12
- "12.0+PTX",
 
13
  "7.5",
14
  "8.0",
15
  "8.6",
 
1
  {
2
  "name": "flashrt-spatiotemporal-layout",
3
+ "id": "_flashrt_spatiotemporal_layout_cuda_5de4768",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
9
  "archs": [
10
  "10.0",
11
  "11.0",
12
+ "12.0",
13
+ "12.1+PTX",
14
  "7.5",
15
  "8.0",
16
  "8.6",
build/torch212-cxx11-cu132-x86_64-linux/__init__.py CHANGED
@@ -21,6 +21,15 @@ def _ncdhw_to_blc_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
21
  return None
22
 
23
 
 
 
 
 
 
 
 
 
 
24
  @torch.library.register_fake(add_op_namespace_prefix("time_unshuffle2_bf16"))
25
  def _time_unshuffle2_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
26
  _check_ncdhw(x, "x")
@@ -58,6 +67,15 @@ def ncdhw_to_blc_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch
58
  return out
59
 
60
 
 
 
 
 
 
 
 
 
 
61
  def time_unshuffle2_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
62
  """Convert BF16 (B, 2C, T, H, W) into (B, C, 2T, H, W)."""
63
 
@@ -85,6 +103,7 @@ def update_cache2_ncdhw_bf16(cur: torch.Tensor, prev: torch.Tensor, out: torch.T
85
 
86
  __all__ = [
87
  "ncdhw_to_blc_bf16",
 
88
  "time_unshuffle2_bf16",
89
  "add_bias_ncdhw_bf16",
90
  "update_cache2_ncdhw_bf16",
 
21
  return None
22
 
23
 
24
+ @torch.library.register_fake(add_op_namespace_prefix("patch_im2col_bf16"))
25
+ def _patch_im2col_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
26
+ if x.dim() != 4 or x.shape[1:] != (224, 224, 3):
27
+ raise RuntimeError("x must have shape (num_views, 224, 224, 3)")
28
+ if out.shape != (x.shape[0] * 256, 588):
29
+ raise RuntimeError("out must have shape (num_views * 256, 588)")
30
+ return None
31
+
32
+
33
  @torch.library.register_fake(add_op_namespace_prefix("time_unshuffle2_bf16"))
34
  def _time_unshuffle2_bf16_fake(x: torch.Tensor, out: torch.Tensor) -> None:
35
  _check_ncdhw(x, "x")
 
67
  return out
68
 
69
 
70
+ def patch_im2col_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
71
+ """Convert BF16 NHWC images into flattened 14x14 patch rows."""
72
+
73
+ if out is None:
74
+ out = torch.empty((x.shape[0] * 256, 588), device=x.device, dtype=x.dtype)
75
+ ops.patch_im2col_bf16(x, out)
76
+ return out
77
+
78
+
79
  def time_unshuffle2_bf16(x: torch.Tensor, out: torch.Tensor | None = None) -> torch.Tensor:
80
  """Convert BF16 (B, 2C, T, H, W) into (B, C, 2T, H, W)."""
81
 
 
103
 
104
  __all__ = [
105
  "ncdhw_to_blc_bf16",
106
+ "patch_im2col_bf16",
107
  "time_unshuffle2_bf16",
108
  "add_bias_ncdhw_bf16",
109
  "update_cache2_ncdhw_bf16",
build/torch212-cxx11-cu132-x86_64-linux/{_flashrt_spatiotemporal_layout_cuda_cf903dd.abi3.so → _flashrt_spatiotemporal_layout_cuda_5de4768.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ff8ee1e1c690aff7cead2ecad593054e5405abd597b9a00b77a4526a7a5995f
3
- size 2522944
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5df2034b32ce3972e68eebd56c25981208b778c3fe6f19f1fce64a47695fa3d8
3
+ size 702464
build/torch212-cxx11-cu132-x86_64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flashrt_spatiotemporal_layout_cuda_cf903dd
3
- ops = torch.ops._flashrt_spatiotemporal_layout_cuda_cf903dd
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flashrt_spatiotemporal_layout_cuda_cf903dd::{op_name}"
 
1
  import torch
2
+ from . import _flashrt_spatiotemporal_layout_cuda_5de4768
3
+ ops = torch.ops._flashrt_spatiotemporal_layout_cuda_5de4768
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flashrt_spatiotemporal_layout_cuda_5de4768::{op_name}"
build/torch212-cxx11-cu132-x86_64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flashrt-spatiotemporal-layout",
3
- "id": "_flashrt_spatiotemporal_layout_cuda_cf903dd",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
@@ -9,7 +9,8 @@
9
  "archs": [
10
  "10.0",
11
  "11.0",
12
- "12.0+PTX",
 
13
  "7.5",
14
  "8.0",
15
  "8.6",
 
1
  {
2
  "name": "flashrt-spatiotemporal-layout",
3
+ "id": "_flashrt_spatiotemporal_layout_cuda_5de4768",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
9
  "archs": [
10
  "10.0",
11
  "11.0",
12
+ "12.0",
13
+ "12.1+PTX",
14
  "7.5",
15
  "8.0",
16
  "8.6",