OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 51 Packages Projects Releases Wiki Activity
Files
b469ed72d0ce791edbc124cd2c4504c37278e30d
pytorch/caffe2/utils
History
Kunhao ZHENG b06c180a32 CUBLAS Flag (CUBLAS_GEMM_DFALT_TENSOR_OP -> CUBLAS_GEMM_DEFAULT_TENSOR_OP) (#100976) 
Looking at the docs https://docs.nvidia.com/cuda/cublas/index.html?highlight=cublasGemmEx#cublasgemmex It seems like the flag should be `CUBLAS_GEMM_DEFAULT_TENSOR_OP`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/100976
Approved by: https://github.com/ezyang
2023-05-10 03:26:31 +00:00
..
hip
…
math
…
threadpool
…
bench_utils.cc
…
bench_utils.h
…
cast_test.cc
…
cast.h
…
cblas.h
…
CMakeLists.txt
S390x tests (#99871)
2023-04-26 21:48:03 +00:00
conversions.h
…
cpu_neon.h
…
cpuid_test.cc
…
cpuid.cc
…
cpuid.h
…
cub_namespace.cuh
…
eigen_utils.h
…
fatal_signal_asan_no_sig_test.cc
…
filler.h
…
fixed_divisor_test.cc
…
fixed_divisor.h
…
GpuAtomics.cuh
…
GpuBitonicSort.cuh
…
GpuDefs.cuh
…
GpuScanUtils.cuh
…
knob_patcher.cc
…
knob_patcher.h
…
knobs_test.cc
…
knobs.cc
…
knobs.h
…
map_utils.h
…
math_cpu.cc
…
math_gpu_test.cc
…
math_gpu.cu
CUBLAS Flag (CUBLAS_GEMM_DFALT_TENSOR_OP -> CUBLAS_GEMM_DEFAULT_TENSOR_OP) (#100976)
2023-05-10 03:26:31 +00:00
math_test.cc
…
math-detail.h
…
math.h
…
murmur_hash3.cc
…
murmur_hash3.h
…
proto_utils_test.cc
…
proto_utils.cc
…
proto_utils.h
…
proto_wrap.cc
…
proto_wrap.h
…
signal_handler.cc
…
signal_handler.h
…
simple_queue_test.cc
…
simple_queue.h
…
smart_tensor_printer_test.cc
…
smart_tensor_printer.cc
…
smart_tensor_printer.h
…
string_utils.cc
…
string_utils.h
…
zmq_helper.h
…