mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-04-20 06:49:15 +00:00
[CK_TILE] naive attn (#1708)
* add reference attention fwd * refactor addresser * update * paged, and i8 reflect-quant * lets call it forward-quant * fix error in decode variation * update naive-attn * fix page table * fix build err
This commit is contained in:
@@ -7,6 +7,7 @@ import copy
|
||||
|
||||
NS = 'ck_tile'
|
||||
OPS = 'ops'
|
||||
REF = 'ref'
|
||||
OPS_COMMON = 'common' # common header will be duplicated into ops/* other module
|
||||
|
||||
HEADER_COMMON = f"""// SPDX-License-Identifier: MIT
|
||||
@@ -29,6 +30,9 @@ class submodule_t:
|
||||
def push(self, f):
|
||||
if len(f.parents) != 1: # ignore ./xxx.hpp
|
||||
mod = get_module(f)
|
||||
# ref is supposed to include one header on demand
|
||||
if mod == REF:
|
||||
return
|
||||
if mod == OPS:
|
||||
if mod not in self.m.keys():
|
||||
self.m[mod] = dict()
|
||||
|
||||
Reference in New Issue
Block a user