Path	Lines of Code
blocksparse/__init__.py	113
blocksparse/conv.py	786
blocksparse/embed.py	30
blocksparse/ewops.py	287
blocksparse/grads.py	185
blocksparse/lstm.py	164
blocksparse/matmul.py	645
blocksparse/nccl.py	310
blocksparse/norms.py	191
blocksparse/optimize.py	228
blocksparse/quantize.py	154
blocksparse/transformer.py	480
blocksparse/utils.py	212
generate_kernels.py	253
setup.py	16
src/batch_norm_op.cc	193
src/batch_norm_op_gpu.cu	393
src/blocksparse_conv_op.cc	384
src/blocksparse_hgemm_cn_128_op_gpu.cu	1112
src/blocksparse_hgemm_cn_64_op_gpu.cu	1425
src/blocksparse_hgemm_nc_op_gpu.cu	1224
src/blocksparse_kernels.cc	62
src/blocksparse_l2_norm_op.cc	507
src/blocksparse_l2_norm_op_gpu.cu	785
src/blocksparse_matmul.h	235
src/blocksparse_matmul_gated_op_gpu.cu	882
src/blocksparse_matmul_op.cc	653
src/blocksparse_matmul_op_gpu.cu	2499
src/bst_hgemm_op_gpu.cu	1100
src/bst_op.cc	481
src/bst_sgemm_op_gpu.cu	414
src/bst_softmax_op_gpu.cu	455
src/cwise_linear_op.cc	161
src/cwise_linear_op_gpu.cu	213
src/dev/spatial_conv.py	389
src/edge_bias_op.cc	182
src/edge_bias_op_gpu.cu	214
src/embedding_op.cc	172
src/embedding_op_gpu.cu	213
src/ew_op.cc	1270
src/ew_op_gpu.cu	1503
src/ew_op_gpu.h	1035
src/gpu_hmma.h	328
src/gpu_types.cc	100
src/gpu_types.h	173
src/layer_norm_cn_op_gpu.cu	695
src/layer_norm_nc_op_gpu.cu	792
src/layer_norm_op.cc	469
src/lstm_op.cc	394
src/lstm_op_gpu.cu	590
src/matmul_op.cc	69
src/matmul_op_gpu.cu	304
src/nccl_op.cc	480
src/optimize_op.cc	747
src/optimize_op_gpu.cu	1030
src/quantize_op.cc	251
src/quantize_op_gpu.cu	216
src/sass/xconv_blocksparse_32x32x16_bprop.sass	737
src/sass/xconv_blocksparse_32x32x16_fprop.sass	737
src/sass/xconv_blocksparse_32x32x32_updat.sass	763
src/sass/xgemm_blocksparse_32x32x32_xprop.sass	655
src/sass/xgemm_blocksparse_32x32x8_updat.sass	674
src/transformer_op.cc	450
src/transformer_op_gpu.cu	884