Skip to content

Commit

Permalink
[Formatting] Apply formating
Browse files Browse the repository at this point in the history
This commit applies formating to rebased code

Signed-off-by: Dmitrii Makarenko <[email protected]>
  • Loading branch information
Devjiu committed Aug 13, 2024
1 parent b647cce commit d71038d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
4 changes: 3 additions & 1 deletion python/test/unit/language/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -1792,6 +1792,7 @@ def kernel(output_ptr, n_elements, BLOCK_SIZE: tl.constexpr):

assert torch.all(output == ref)


@pytest.mark.cpu
@pytest.mark.interpreter
@pytest.mark.parametrize("num_ctas", num_ctas_list)
Expand Down Expand Up @@ -5389,7 +5390,8 @@ def matmul_kernel( #
@pytest.mark.cpu
@pytest.mark.interpreter
@pytest.mark.parametrize("M, N, K", [(128, 256, 256)])
@pytest.mark.parametrize("BLOCK_M, BLOCK_N, BLOCK_K", [(128, 256, 128), (64, 64, 64)] if not is_cpu() else [(32, 32, 128), (32, 32, 32)])
@pytest.mark.parametrize("BLOCK_M, BLOCK_N, BLOCK_K", [(128, 256, 128),
(64, 64, 64)] if not is_cpu() else [(32, 32, 128), (32, 32, 32)])
@pytest.mark.parametrize("in_type_str", ['float8e5', 'float8e4nv', 'float8e4b15'])
@pytest.mark.parametrize("low_precision_acc", [0, 32, 64, 128])
def test_dot_max_num_imprecise_acc(M, N, K, BLOCK_M, BLOCK_N, BLOCK_K, in_type_str, low_precision_acc, device):
Expand Down
1 change: 1 addition & 0 deletions third_party/cpu/backend/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
from triton.runtime.build import _build
import triton.backends.cpu.driver as cpu_driver


def min_dot_size(target: GPUTarget):
# Other architectures will only support 16,16,16
return lambda lhsType, rhsType: (4, 4, 4)
Expand Down

0 comments on commit d71038d

Please sign in to comment.