Skip to content

Instantly share code, notes, and snippets.

@jamesjwu
Last active May 28, 2024 22:20
Show Gist options
  • Save jamesjwu/a76490e55e27641697cc60f172cab908 to your computer and use it in GitHub Desktop.
Save jamesjwu/a76490e55e27641697cc60f172cab908 to your computer and use it in GitHub Desktop.
Triton compilation failed: triton_poi_fused_0
defb844cc147', 'are_deterministic_algorithms_enabled': False, 'assert_indirect_indexing': True, 'autotune_local_cache': True, 'autotune_pointwise': True, 'autotune_remote_cache': False, 'force_disable_caches': False, 'dynamic_scale_rblock': True, 'max_autotune': False, 'max_autotune_pointwise': False, 'min_split_scan_rblock': 256, 'spill_threshold': 16, 'store_cubin': False},
min_elem_per_thread=0
)
@triton.jit
def triton_(in_ptr0, out_ptr0, ynumel, xnumel, YBLOCK : tl.constexpr, XBLOCK : tl.constexpr):
ynumel = 192
xnumel = 49
yoffset = tl.program_id(1) * (tl.program_id(2) + 1) * YBLOCK
yindex = yoffset + tl.arange(0, YBLOCK)[None, :]
ymask = yindex < ynumel
xoffset = tl.program_id(0) * XBLOCK
xindex = xoffset + tl.arange(0, XBLOCK)[:, None]
xmask = xindex < xnumel
x2 = xindex
y3 = yindex
y0 = yindex % 3
y1 = (yindex // 3)
tmp0 = tl.load(in_ptr0 + (x2 + (49*y3)), xmask & ymask, eviction_policy='evict_last')
tl.store(out_ptr0 + (y0 + (3*x2) + (147*y1)), tmp0, xmask & ymask)
metadata: {'signature': {0: '*fp32', 1: '*fp32', 2: 'i32', 3: 'i32'}, 'device': 0, 'constants': {5: 32, 4: 32}, 'configs': [AttrsDescriptor(divisible_by_16=(0, 1, 2), equal_to_1=())], 'device_type': 'cuda', 'num_warps': 4, 'num_stages': 1, 'debug': True, 'cc': 86}
Traceback (most recent call last):
File "/__w/_temp/conda_environment_9276673568/lib/python3.11/site-packages/torch/_inductor/runtime/triton_heuristics.py", line 430, in _precompile_config
binary = triton.compile(*compile_args, **compile_kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/__w/_temp/conda_environment_9276673568/lib/python3.11/site-packages/triton/compiler/compiler.py", line 233, in compile
key = f"{triton_key()}-{src.hash()}-{backend.hash()}-{options.hash()}-{str(sorted(get_env_vars().items()))}"
^^^^^^^^^^
File "/__w/_temp/conda_environment_9276673568/lib/python3.11/site-packages/triton/compiler/compiler.py", line 108, in hash
key = f"{self.fn.cache_key}-{self.attrs.hash()}-{sorted_sig}-{sorted_constants}"
^^^^^^^^^^^^^^^^^
File "/__w/_temp/conda_environment_9276673568/lib/python3.11/site-packages/triton/runtime/jit.py", line 514, in cache_key
dependencies_finder.visit(self.parse())
^^^^^^^^^^^^
File "/__w/_temp/conda_environment_9276673568/lib/python3.11/site-packages/triton/runtime/jit.py", line 549, in parse
tree = ast.parse(self.src)
^^^^^^^^^^^^^^^^^^^
File "/__w/_temp/conda_environment_9276673568/lib/python3.11/ast.py", line 50, in parse
return compile(source, filename, mode, flags,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "<unknown>", line 1
defb844cc147', 'are_deterministic_algorithms_enabled': False, 'assert_indirect_indexing': True, 'autotune_local_cache': True, 'autotune_pointwise': True, 'autotune_remote_cache': False, 'force_disable_caches': False, 'dynamic_scale_rblock': True, 'max_autotune': False, 'max_autotune_pointwise': False, 'min_split_scan_rblock': 256, 'spill_threshold': 16, 'store_cubin': False},
^
SyntaxError: unterminated string literal (detected at line 1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment