ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(copyright): update copyright header for example directory (#3273)

Browse Source 
* chore(copyright): update copyright header for codegen directory

* chore(copyright): update copyright header for example directory
This commit is contained in:
Aviral Goel
2025-11-25 06:02:41 +04:00
committed by GitHub
parent 229d43ea0c
commit d85f065b15
729 changed files with 1077 additions and 1025 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
example/ck_tile/02_layernorm2d/generate.py
View File

@@ -1,7 +1,8 @@
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
# SPDX-License-Identifier: MIT
# Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
# generate kernel instances to speed up compilation
import argparse
from enum import IntEnum
from pathlib import Path

3
example/ck_tile/02_layernorm2d/layernorm2d_fwd.cpp
View File

@@ -1,3 +1,6 @@
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#include "ck_tile/host.hpp"
#include "layernorm2d_fwd.hpp"
#include "ck_tile/utility/json_dump.hpp"

2
example/ck_tile/02_layernorm2d/layernorm2d_fwd.hpp
View File

@@ -1,5 +1,5 @@
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once

3
example/ck_tile/02_layernorm2d/script/perf_test.sh
View File

@@ -1,4 +1,7 @@
#!/bin/sh
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
# SPDX-License-Identifier: MIT
EXE="$(find . -name tile_example_layernorm2d_fwd -type f | head -n 1)"
$EXE -m=1 -n=1 -e=1e-12 -v=1 -prec_i=bf16 -repeat=1000

3
example/ck_tile/02_layernorm2d/script/smoke_test.sh
View File

@@ -1,4 +1,7 @@
#!/bin/sh
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
# SPDX-License-Identifier: MIT
EXE="$(find . -name tile_example_layernorm2d_fwd -type f | head -n 1)"
for fquant in "" "-fquant=1 -prec_o=int8" "-fquant=1 -prec_o=fp8"; do