ninetoothed 0.15.0__py3-none-any.whl → 0.15.1__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 (8) hide show
  1. ninetoothed/aot.py +2 -2
  2. ninetoothed/generation.py +1 -0
  3. ninetoothed/tensor.py +1 -1
  4. ninetoothed/visualization.py +10 -4
  5. {ninetoothed-0.15.0.dist-info → ninetoothed-0.15.1.dist-info}/METADATA +1 -1
  6. {ninetoothed-0.15.0.dist-info → ninetoothed-0.15.1.dist-info}/RECORD +8 -8
  7. {ninetoothed-0.15.0.dist-info → ninetoothed-0.15.1.dist-info}/WHEEL +0 -0
  8. {ninetoothed-0.15.0.dist-info → ninetoothed-0.15.1.dist-info}/licenses/LICENSE +0 -0
ninetoothed/aot.py CHANGED
@@ -4,7 +4,7 @@ import subprocess
4
4
  import tempfile
5
5
  import uuid
6
6
 
7
- from ninetoothed.dtype import int64, uint64
7
+ from ninetoothed.dtype import int64
8
8
  from ninetoothed.generation import CACHE_DIR, CodeGenerator
9
9
  from ninetoothed.tensor import Tensor
10
10
 
@@ -55,7 +55,7 @@ def _aot(func, caller, kernel_name, num_warps, num_stages):
55
55
 
56
56
  param_types.append(f"*{dtype}")
57
57
  elif Tensor.size_pattern().fullmatch(param):
58
- param_types.append(uint64)
58
+ param_types.append(int64)
59
59
  elif Tensor.stride_pattern().fullmatch(param):
60
60
  param_types.append(int64)
61
61
 
ninetoothed/generation.py CHANGED
@@ -82,6 +82,7 @@ class CodeGenerator(ast.NodeTransformer):
82
82
  dependencies = _find_dependencies(func)
83
83
  source = "\n\n".join((unparsed, dependencies)).strip()
84
84
  source = source.replace(func.__name__, kernel_name)
85
+ source += "\n"
85
86
 
86
87
  if prettify:
87
88
  for original, simplified in name_collector.simplified_names.items():
ninetoothed/tensor.py CHANGED
@@ -146,7 +146,7 @@ class Tensor:
146
146
  )
147
147
  outer_shape.append(new_size)
148
148
 
149
- new_stride = self_stride * stride // spacing
149
+ new_stride = self_stride * stride
150
150
  outer_strides.append(new_stride)
151
151
 
152
152
  inner_shape.append(tile_size)
ninetoothed/visualization.py CHANGED
@@ -118,10 +118,16 @@ def _visualize_unit_square(ax, x, y, color):
118
118
 
119
119
 
120
120
  def _visualize_rect(ax, width, height, x, y, color):
121
- pos_x, pos_y = zip(*_verts_of_rect(width, height, x, y))
122
-
123
- ax.fill(pos_x, pos_y, color)
124
- ax.plot(pos_x + (pos_x[0],), pos_y + (pos_y[0],), "k")
121
+ ax.add_patch(
122
+ plt.Rectangle(
123
+ (x, y),
124
+ width,
125
+ height,
126
+ edgecolor="k",
127
+ facecolor=color,
128
+ linewidth=plt.rcParams["lines.linewidth"],
129
+ )
130
+ )
125
131
 
126
132
 
127
133
  def _verts_of_rect(width, height, x, y):
{ninetoothed-0.15.0.dist-info → ninetoothed-0.15.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ninetoothed
3
- Version: 0.15.0
3
+ Version: 0.15.1
4
4
  Summary: A domain-specific language based on Triton but providing higher-level abstraction.
5
5
  Project-URL: Homepage, https://github.com/InfiniTensor/ninetoothed
6
6
  Project-URL: Issues, https://github.com/InfiniTensor/ninetoothed/issues
{ninetoothed-0.15.0.dist-info → ninetoothed-0.15.1.dist-info}/RECORD RENAMED
@@ -1,17 +1,17 @@
1
1
  ninetoothed/__init__.py,sha256=zGaZiUzwJZ2jfwLxp7lT8ll_V5ngP5QYrfVbapftbCY,522
2
- ninetoothed/aot.py,sha256=1hzl4-6MqscB4tDMqmLCOTlyzsYkbY20EnmDgHO8hU4,6137
2
+ ninetoothed/aot.py,sha256=5P9s-KAA7xNNdK8_fbCZEIteQlbaB_1wOl8_rEBQg9U,6128
3
3
  ninetoothed/cudaifier.py,sha256=5ylMr1q0B9NwbeXkpCu3o2nMGpDfh65nAQ0Az_qMQuI,877
4
4
  ninetoothed/dtype.py,sha256=-0iBleay5gYA4wtT3l17QjCesr7g26M6CSfhNJdI3k4,165
5
- ninetoothed/generation.py,sha256=QHrK7DOuJo5wEV-5HAqu_e-suuD4TPPNmCzrUMRYF2w,30940
5
+ ninetoothed/generation.py,sha256=Gmeh9OPmWZmF9CUY-UIIBPi-SOjFCxZjvXNwqX3uD84,30963
6
6
  ninetoothed/jit.py,sha256=0MFbFIODtw-bxuOC7WByxiVtQMeyvZkoDxvfAZ9rIFQ,2120
7
7
  ninetoothed/language.py,sha256=YwjlBENmmKPTnhaQ2uYbj5MwzrCAT7MLJ6VkQ6NeXJE,504
8
8
  ninetoothed/make.py,sha256=wRr3JwGt5E2OCquq_nzBZljdW-AJPOqH49cM08gwl4A,1287
9
9
  ninetoothed/naming.py,sha256=Fl0x4eDRStTpkXjJg6179ErEnY7bR5Qi0AT6RX9C3fU,951
10
10
  ninetoothed/symbol.py,sha256=UpGmx_jvaDtowADnp1DwYC3fvBXSiaMiYpU-ewkVo50,5261
11
- ninetoothed/tensor.py,sha256=W1XY8_vaYmszX4lIWuas-ZKGbbdEZU7Z5h1A4FBXDXg,14358
11
+ ninetoothed/tensor.py,sha256=ByTnoeqxD9lXprvy1DDp5L-zU2up52-jop9AAUrSTYk,14347
12
12
  ninetoothed/torchifier.py,sha256=aDijK5UOwK2oLXDHgDo8M959rJclEI0lcfaPr7GQTXY,1012
13
- ninetoothed/visualization.py,sha256=IZ7iTT4dl5_JFbO-WfSWPFWpgkyPr4nylwhSZVy8gss,3601
14
- ninetoothed-0.15.0.dist-info/METADATA,sha256=UmND-TBDf7vrdii9dhiOZmiZBzrcG8-xEniThTSikqM,7311
15
- ninetoothed-0.15.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
- ninetoothed-0.15.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
17
- ninetoothed-0.15.0.dist-info/RECORD,,
13
+ ninetoothed/visualization.py,sha256=zlMH-0WplaboePGzcbpcj4UovpX0k2r4SysSPsNS4r4,3674
14
+ ninetoothed-0.15.1.dist-info/METADATA,sha256=6RA1-6fYFfSTJnnwsRoVb4yIRrn4kfLhN47GNvmGji0,7311
15
+ ninetoothed-0.15.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
+ ninetoothed-0.15.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
17
+ ninetoothed-0.15.1.dist-info/RECORD,,