get_num_tblis_threads to unexported get_num_threads

Author: lkdvos

src/TensorOperationsTBLIS.jl

@@ -16,10 +16,9 @@ using .LibTBLIS
 using .LibTBLIS: LibTBLIS, len_type, stride_type
 
 export TBLIS
-export get_num_tblis_threads, set_num_tblis_threads
 
-get_num_tblis_threads() = convert(Int, LibTBLIS.tblis_get_num_threads())
-set_num_tblis_threads(n) = LibTBLIS.tblis_set_num_threads(convert(Cuint, n))
+get_num_threads() = convert(Int, LibTBLIS.tblis_get_num_threads())
+set_num_threads(n) = LibTBLIS.tblis_set_num_threads(convert(Cuint, n))
 
 # TensorOperations
 #------------------
@@ -34,8 +33,8 @@ or BLAS.
 struct TBLIS <: TensorOperations.AbstractBackend end
 
 Base.@deprecate(tblisBackend(), TBLIS())
-Base.@deprecate(tblis_get_num_threads(), get_num_tblis_threads())
-Base.@deprecate(tblis_set_num_threads(n), set_num_tblis_threads(n))
+Base.@deprecate(tblis_get_num_threads(), get_num_threads())
+Base.@deprecate(tblis_set_num_threads(n), set_num_threads(n))
 
 include("strided.jl")
 

test/runtests.jl

@@ -5,16 +5,16 @@ using Test, Random, LinearAlgebra
 Random.seed!(1234567)
 
 @testset "TensorOperationsTBLIS.jl" begin
-    set_num_tblis_threads(1)
+    TensorOperationsTBLIS.set_num_threads(1)
     @testset "method syntax" verbose = true begin
         include("methods.jl")
     end
 
     @test get_num_tblis_threads() == 1
-    set_num_tblis_threads(2)
+    TensorOperationsTBLIS.set_num_threads(2)
     @testset "macro with index notation" verbose = true begin
         include("tensor.jl")
     end
 
-    @test get_num_tblis_threads() == 2
+    @test TensorOperationsTBLIS.get_num_threads() == 2
 end