using TropicalGEMM

GiggleLiu · GiggleLiu · commit 066c4d39d8c1 · 2023-06-29T22:42:46.000+08:00
diff --git a/Project.toml b/Project.toml
@@ -10,13 +10,15 @@ DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 OMEinsum = "ebe7aa44-baf0-506c-a96f-8464559b3922"
 Requires = "ae029012-a4dd-5104-9daa-d747884805df"
+TropicalGEMM = "a4ad3063-64a7-4bad-8738-34ed09bc0236"
 TropicalNumbers = "b3a74e9c-7526-4576-a4eb-79c0d4c32334"
 
 [compat]
-CUDA = "3"
+CUDA = "4"
 DocStringExtensions = "0.8.6, 0.9"
 OMEinsum = "0.7"
 Requires = "1"
+TropicalGEMM = "0.1"
 TropicalNumbers = "0.5.4"
 julia = "1.3"
 
diff --git a/src/TensorInference.jl b/src/TensorInference.jl
@@ -3,6 +3,7 @@ module TensorInference
 using OMEinsum, LinearAlgebra
 using DocStringExtensions, TropicalNumbers
 using Artifacts
+using TropicalGEMM
 
 # reexport OMEinsum functions
 export RescaledArray
