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

[examples/python/tensorflow] fixed ABI compatibility issue in JIT

Browse Source 
destructor
This commit is contained in:
Philippe Tillet
2019-05-01 13:38:56 -04:00
parent 7b6efc0463
commit 55866f1ef6
5 changed files with 47 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/python/tensorflow/blocksparse.py
View File

@@ -3,7 +3,7 @@ import tensorflow as tf
import numpy as np
data_files_path = tf.resource_loader.get_data_files_path()
library_dir = '/home/philippe/development/triton/build/examples/python/tensorflow'
library_dir = '/home/philippe/Development/triton/build/examples/python/tensorflow'
module = tf.load_op_library(os.path.join(library_dir, 'libtf_blocksparse.so'))
M, N, K = 512, 512, 512