gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[general] fixed some warnings

Browse Source
This commit is contained in:
Philippe Tillet
2019-08-18 14:08:57 -07:00
parent c05445d001
commit 81571246cf
22 changed files with 75 additions and 365 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/examples/dot.py
View File

@@ -11,7 +11,8 @@ void matmul(restrict read_only align(16) half *A,
restrict read_only align(16) half *B,
restrict read_only align(16) half *C,
int M, int N, int K,
multiple_of(8) int lda, multiple_of(8) int ldb, int ldc) {
multiple_of(8) int lda, multiple_of(8) int ldb, int ldc)
{
int ridx = get_program_id(0);
int ridy = get_program_id(1);
int rxa[TM] = ridx * TM + (0 ... TM);

4
python/triton/ops.py
View File

@@ -17,8 +17,8 @@ import tensorflow as tf
extra_ops = tf.load_op_library('/home/philippe/development/triton/python/build/lib.linux-x86_64-3.6/libextra_tf_ops.so')
def make_bindings(src, outputs, grids):
return libtriton.make_tensorflow_src(src, outputs, grids)
def make_bindings(src, out, grid):
return libtriton.make_tensorflow_src(src, out, grid)
def make_cache_path(src):
md5 = hashlib.sha1(src.encode())